This commit is contained in:
Glenn Jocher 2019-02-20 18:41:31 +01:00
parent f8c675dbc0
commit 0b971eddff
1 changed files with 1 additions and 1 deletions

View File

@ -162,7 +162,7 @@ def train(
if best_loss == loss_per_target: if best_loss == loss_per_target:
os.system('cp ' + latest + ' ' + best) os.system('cp ' + latest + ' ' + best)
# Save backup weights every 5 epochs # Save backup weights every 5 epochs (optional)
# if (epoch > 0) & (epoch % 5 == 0): # if (epoch > 0) & (epoch % 5 == 0):
# os.system('cp ' + latest + ' ' + os.path.join(weights, 'backup{}.pt'.format(epoch))) # os.system('cp ' + latest + ' ' + os.path.join(weights, 'backup{}.pt'.format(epoch)))