Solved by anantguptadbl in post #2 
              
              
                @Artemisia_Jonath 
The following statement changes your channels from 64 to 1 

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