diff --git a/train.py b/train.py index c298692b7335..dc93c22d621a 100644 --- 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,