bug fix on training avg loss printing and computing

This commit is contained in:
Eren Golge committed 2018-04-12 05:57:52 -07:00
1 parent 6c2f52b071
commit bc90050ee9
3 files changed
+17 -20

No files matched your search

+1 -1
View File
@@ -103,7 +103,7 @@ def save_best_model(model, optimizer, model_loss, best_loss, out_path,
best_loss = model_loss
bestmodel_path = 'best_model.pth.tar'
bestmodel_path = os.path.join(out_path, bestmodel_path)
print("\n | > Best model saving with loss {0:.2f} : {1:}".format(
print(" | > Best model saving with loss {0:.2f} : {1:}".format(
model_loss, bestmodel_path))
torch.save(state, bestmodel_path)
return best_loss