-for m in c:
- m = nn.Sequential(*[ nn.Conv1d(1, 1, l[0], l[1]) for l in m ])
- print(m)
- print(x.size(), m(x).size())
+ x = torch.rand(1, 1, 64)
+
+ for m in c:
+ model = nn.Sequential(*[ nn.Conv1d(1, 1, l[0], l[1]) for l in m ])
+ print(model)
+ print(x.size(), model(x).size())