Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-02 16:33:52 +02:00
parent 658f2a4576
commit d526ce0d11
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
train.py
View File

@ -180,15 +180,16 @@ def train(
model) is nn.parallel.DistributedDataParallel else model.state_dict(), model) is nn.parallel.DistributedDataParallel else model.state_dict(),
'optimizer': optimizer.state_dict()} 'optimizer': optimizer.state_dict()}
torch.save(checkpoint, latest) torch.save(checkpoint, latest)
del checkpoint
# Save best checkpoint # Save best checkpoint
if best_loss == mloss['total']: if best_loss == mloss['total']:
os.system('cp ' + latest + ' ' + best) torch.save(checkpoint, best)
# Save backup weights every 10 epochs (optional) # Save backup weights every 10 epochs (optional)
if epoch > 0 and epoch % 10 == 0: if epoch > 0 and epoch % 10 == 0:
os.system('cp ' + latest + ' ' + weights + 'backup%g.pt' % epoch) torch.save(checkpoint, weights + 'backup%g.pt' % epoch)
del checkpoint
# Calculate mAP # Calculate mAP
with torch.no_grad(): with torch.no_grad():