bug fix on training avg loss printing and computing

2026-09-09 11:16:00 +08:00 · 2018-04-12 05:57:52 -07:00
parent 6c2f52b071
commit bc90050ee9
3 changed files with 17 additions and 20 deletions
@@ -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