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

updates

This commit is contained in:
Glenn Jocher 2019-07-24 13:32:14 +02:00
parent 5a34d3c9ab
commit a8596c6af4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -317,7 +317,7 @@ def train(cfg,
# Report time
print('%g epochs completed in %.3f hours.' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))
del model, optimizer, loss, dataset, dataloader, scaled_loss, scheduler
del model, optimizer, loss, dataset, dataloader, scheduler
torch.cuda.empty_cache()
return results