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

Update train.py for `val.run(half=amp)` (#8804)

Disable FP16 validation if AMP checks fail or amp=False.
上级 1e89807d
......@@ -367,6 +367,7 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
results, maps, _ = val.run(data_dict,
batch_size=batch_size // WORLD_SIZE * 2,
imgsz=imgsz,
half=amp,
model=ema.ema,
single_cls=single_cls,
dataloader=val_loader,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论