The output size of ConvTranspose2d differs from the expected output size

5

There is output_size argument in nn.ConvTranspose2d.forward function. Just provide desired output shape during forward pass.

2 Likes