@ptrblck  I have another question while training the autoencoder on my dataset,at the end of the training ,the output came out to be in the following format
[ 70.6182,  70.8503,  70.6278,  ..., 243.5364, 330.0468,   5.6832]

But at evaluation stage,when I run the autoencoder on a single image and a random vector of 3x1,the autoencoder generates the following ouptut
tensor([[3.2402e+16, 3.4111e+16, 3.2839e+16,  ..., 4.4640e+16, 4.4089e+16,
         7.7656e+15]], device='cuda:0', dtype=torch.float64)

As you can see the values are not close to the ones obtained during training,any ideas why this is happening.