mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
allow loss to be used for early stopping (#127)
This commit is contained in:
1 parent
bdd86087e6
commit
308239cef0
1 file changed
+1
-1
@@ -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),
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user