Hi!
My input is a batch of size: [10, 1, 257, 65]
I might have some problem with the Conv2d dimensions.
Here is attached the model. Please help me solve it.
Thanks.
class SimpleCNN(torch.nn.Module):
def init(self):
super(SimpleCNN, self).init()
#Input channels = 1, output channels = 16
self.layer1 = nn.Sequential(
nn.Conv2d(1, 16, kernel_size=3, stride=2),
nn.ReLU(),
nn.Conv2d(16, 32,kernel_size=3, stride=2),
nn.ReLU(),
nn.Conv2d(32, 32, kernel_size=3, stride=2),
nn.ReLU(),
nn.Conv2d(32, 32, kernel_size=3, stride=3),
nn.ReLU(),
nn.Conv2d(32, 64, kernel_size=3, stride=2),
nn.ReLU(),
nn.Conv2d(64, 64, kernel_size=3, stride=2)
)
self.fc1 = nn.Linear(64, 166)
self.sig = nn.Sigmoid()
def forward(self, x):
out = self.layer1(x)
out = self.fc1(out)
out = out.view(out.size(0), -1)
out = self.sig(out)
return out