pourya_farzi:

I used nn.Conv2d since I decided to add more channels for the next test


The choice of the type of convolution layer does not depend on the number of channels. Can you post the definitions of sefl.convA1 and self.convB1?
Sorry, I don’t know what this RuntimeError means.