mirror of
https://github.com/wassname/TTS.git
synced 2026-10-29 12:51:00 +08:00
bug fix on training avg loss printing and computing
This commit is contained in:
1 parent
6c2f52b071
commit
bc90050ee9
3 files changed
+17
-20
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user