I’m working on a pytorch implementation of U-net (https://lmb.informatik.uni-freiburg.de/people/ronneber/u-net/ 1).
For up-convolution I’m using nn.ConvTranspose2d. During the training procedure, after some iterations, ConvTranspose2d is returning arrays with all the elements set to ‘nan’. Why is that happening?
Configurations of the up-conv layer: ConvTranspose2d(1024, 512, kernel_size=(2, 2), stride=(2, 2))