updates
This commit is contained in:
parent
a660211733
commit
54d1da904c
2
train.py
2
train.py
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue