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

Disable FP16 validation if AMP checks fail or amp=False.
ctjanuhowski · Sep 8, 2022 · 9337b19 · 9337b19
1 parent 07711f1
commit 9337b19
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -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,