Thanks, I figured out that I should have done some preprocessing. However, I would like to ask you something else. Is there a way to use a max_pool or a conv2d with stride 1/2? If I try to use 1/2 as an argument I get an error saying that I can only use int. I am trying to increase the image size for the decoding part of the network.