This commit is contained in:
Glenn Jocher 2018-08-27 00:00:25 +02:00
parent a660211733
commit 54d1da904c
1 changed files with 1 additions and 1 deletions

View File

@ -178,7 +178,7 @@ def main(opt):
os.system('cp checkpoints/latest.pt checkpoints/best.pt')
# Save backup checkpoint
if (epoch > 0) & (epoch % 10 == 0):
if (epoch > 0) & (epoch % 5 == 0):
os.system('cp checkpoints/latest.pt checkpoints/backup' + str(epoch) + '.pt')
# Save final model