updates
This commit is contained in:
parent
f8c675dbc0
commit
0b971eddff
2
train.py
2
train.py
|
@ -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)))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue