search

L1aoXingyu / pytorch-beginner

pytorch tutorial for beginners
3k stars 1.09k forks source link

error calculation #5

Closed varghesealex90 closed 7 years ago

varghesealex90 commented 7 years ago

In the autoencoder example, the values printed i.e. print('epoch [{}/{}], loss:{:.4f}' .format(epoch + 1, num_epochs, loss.data[0])) this is the error associated with the final batch of the data right ?

L1aoXingyu commented 7 years ago

yes, you can compute the average loss by storing all the loss.