Hey, thanks!
Solved the problem, but I got another one:
“conv_transpose2d(): argument ‘stride’ must be tuple of ints, but found element of type float at pos 1”
Is this because the tensor elements are in float type?
The configuration of my CNN layers is in this layout:
# conv1
          nn.Conv2d(in_channels = 1, out_channels = 64, kernel_size = 3, stride = 1, padding = 1, dilation = 1),
          nn.ReLU(),
          nn.Conv2d(in_channels = 64, out_channels = 64, kernel_size = 3, stride = 2, padding = 1, dilation = 1),
          nn.ReLU(),
          nn.BatchNorm2d(num_features = 64),