pytorch-beginner icon indicating copy to clipboard operation
pytorch-beginner copied to clipboard

Loss and training

Open Jeanselme opened this issue 6 years ago • 0 comments

Hi, I think that there is a few mistakes in the simple and convolutional autoencoders :

  • The displayed loss is the loss on the last image of the epoch instead of the loss over the whole epoch
  • The autoencoder is not tested on the test dataset
  • The autoencoder is never in "train" or "test" mode Is it normal ?

Jeanselme avatar Jun 22 '18 21:06 Jeanselme