mirror of https://github.com/coqui-ai/TTS.git
small print formatting
This commit is contained in:
parent
36efaa789d
commit
22dcc4f7d0
|
@ -123,7 +123,7 @@ def save_best_model(model, optimizer, model_loss, best_loss, out_path,
|
||||||
best_loss = model_loss
|
best_loss = model_loss
|
||||||
bestmodel_path = 'best_model.pth.tar'
|
bestmodel_path = 'best_model.pth.tar'
|
||||||
bestmodel_path = os.path.join(out_path, bestmodel_path)
|
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))
|
model_loss, bestmodel_path))
|
||||||
torch.save(state, bestmodel_path)
|
torch.save(state, bestmodel_path)
|
||||||
return best_loss
|
return best_loss
|
||||||
|
|
Loading…
Reference in New Issue