@Artemisia_Jonath
The following statement changes your channels from 64 to 1



 Artemisia_Jonath:

x = x.view(-1,1,32,32)


Due to this the following statement fails as it expects 64 channels
self.filt1 = nn.Conv2d(64, 64, 2, 2, bias=False, padding=1)