зеркало из https://github.com/mozilla/TTS.git
logging change
This commit is contained in:
Родитель
32de8a579b
Коммит
8ae1ef72fc
|
@ -82,7 +82,7 @@ def save_checkpoint(model, optimizer, model_loss, out_path,
|
|||
current_step, epoch):
|
||||
checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step)
|
||||
checkpoint_path = os.path.join(out_path, checkpoint_path)
|
||||
print("\n | > Checkpoint saving : {}".format(checkpoint_path))
|
||||
print(" | | > Checkpoint saving : {}".format(checkpoint_path))
|
||||
|
||||
new_state_dict = _trim_model_state_dict(model.state_dict())
|
||||
state = {'model': new_state_dict,
|
||||
|
|
Загрузка…
Ссылка в новой задаче