Unverified 提交 dd7f0b7e authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

上级 683cefea
...@@ -419,7 +419,7 @@ def train(hyp, opt, device, tb_writer=None): ...@@ -419,7 +419,7 @@ def train(hyp, opt, device, tb_writer=None):
# Test best.pt # Test best.pt
logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) 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 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, results, _, _ = test.test(opt.data,
batch_size=batch_size * 2, batch_size=batch_size * 2,
imgsz=imgsz_test, imgsz=imgsz_test,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论