Solved by vmirly1 in post #2 
              
              
                There are a couple of things here. 


First of all, the batch-size should not be given as input size to the Linear layer. The input size is independent of batch-size. 


the linear cannot be placed in the same Sequential function right after the conv layers. The reason is the input to linear needs t…