Solved by ptrblck in post #2 
              
              
                I assume that ae_model is an instance of ConvAutoencoder. 
If that’s the case, note that self.modelA will only use the child modules of ConvAutoencoder and will not apply the functional API calls in its forward such as torch.relu and torch.sigmoid and will also use the max pooling layer once. 
nn.Se…