Fix bug with resume (ultralytics#8912)

* Fix bug with resume * restore del on resume=false * Update train.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ctjanuhowski · Sep 8, 2022 · 996484a · 996484a
1 parent 4f64de6
commit 996484a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -167,7 +167,8 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
     # Resume
     best_fitness, start_epoch = 0.0, 0
     if pretrained:
-        best_fitness, start_epoch, epochs = smart_resume(ckpt, optimizer, ema, weights, epochs, resume)
+        if resume:
+            best_fitness, start_epoch, epochs = smart_resume(ckpt, optimizer, ema, weights, epochs, resume)
         del ckpt, csd
 
     # DP mode