checkpoint log

This commit is contained in:
erogol 2020-04-23 14:24:38 +02:00
parent 0e7ecca33f
commit d5093bf6fb
1 changed files with 1 additions and 1 deletions

View File

@ -113,7 +113,7 @@ def save_checkpoint(model, optimizer, optimizer_st, 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(" | | > Checkpoint saving : {}".format(checkpoint_path))
print(" > CHECKPOINT : {}".format(checkpoint_path))
new_state_dict = model.state_dict()
state = {