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