small print formatting

This commit is contained in:
Eren Golge 2018-11-22 17:03:53 +01:00
parent 36efaa789d
commit 22dcc4f7d0
1 changed files with 1 additions and 1 deletions

View File

@ -123,7 +123,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(" | > Best model saving with loss {0:.2f} : {1:}".format(
print(" | > Best model saving with loss {0:.5f} : {1:}".format(
model_loss, bestmodel_path))
torch.save(state, bestmodel_path)
return best_loss