allow loss to be used for early stopping (#127)

2019-08-16 11:58:44 -04:00 · 2019-08-16 11:58:44 -04:00 · 308239cef0
parent bdd86087e6
commit 308239cef0
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -325,7 +325,7 @@ class Trainer(TrainerIO):
    @property
    def __tng_tqdm_dic(self):
        tqdm_dic = {
-            'tng_loss': '{0:.3f}'.format(self.avg_loss),
+            'loss': '{0:.3f}'.format(self.avg_loss),
            'epoch': '{}'.format(self.current_epoch),
            'batch_nb': '{}'.format(self.batch_nb),
        }