davinci26 / Pytorch-Playground

Pytorch is lit 💥 💥
1 stars 0 forks source link

Feat: Validation error during training #1

Closed davinci26 closed 5 years ago

davinci26 commented 5 years ago

Scenario

The user wants to log and view on console output both training and validation error during training. With this feature the user can know if the model is overfitting or if the model is under trained.

Expected output

During training the following log is available [{Epoch}, {Batch}] Training loss: {loss_train} Validation loss: {loss_validation}