wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-26 13:37:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

no validation for now

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2019-12-17 13:47:07 +01:00
parent 2f1ae46ce8
commit aa5d3962b9
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/trainer.py
+1 -2
View File
@@ -50,8 +50,7 @@ class Trainer:
avg_epoch_loss += loss.item()
it.set_postfix(
ordered_dict={
("" if is_training else "validation_")
+ "avg_epoch_loss": avg_epoch_loss/batch_no
"avg_epoch_loss": avg_epoch_loss/batch_no
},
refresh=False,
)