Fix TypeError: 'PosixPath' object is not iterable (ultralytics#3285)

fhkiel-mlaip · May 21, 2021 · 7cef202 · 7cef202
1 parent e1479ab
commit 7cef202
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -419,7 +419,7 @@ def train(hyp, opt, device, tb_writer=None):
         # Test best.pt
         logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))
         if opt.data.endswith('coco.yaml') and nc == 80:  # if COCO
-            for m in (last, best) if best.exists() else (last):  # speed, mAP tests
+            for m in [last, best] if best.exists() else [last]:  # speed, mAP tests
                 results, _, _ = test.test(opt.data,
                                           batch_size=batch_size * 2,
                                           imgsz=imgsz_test,