backbone as FP16, save default to FP32

This commit is contained in:
Glenn Jocher committed 2020-06-18 00:13:18 -07:00
1 parent d9b64c27c2
commit cce95e744d
2 files changed
+2 -3

No files matched your search

+1 -1
View File
@@ -332,7 +332,7 @@ def train(hyp):
ckpt = {'epoch': epoch,
'best_fitness': best_fitness,
'training_results': f.read(),
'model': ema.ema.module.half() if hasattr(model, 'module') else ema.ema.half(),
'model': ema.ema.module if hasattr(model, 'module') else ema.ema,
'optimizer': None if final_epoch else optimizer.state_dict()}
# Save last, best and delete