From 225e6e2b25adf597a0a74bac8b4b92f449025d70 Mon Sep 17 00:00:00 2001 From: Laughing <61612323+Laughing-q@users.noreply.github.com> Date: Wed, 18 Sep 2024 16:09:56 +0800 Subject: [PATCH] Use `trainer.amp` to determine FP16 validation (#16333) Co-authored-by: Glenn Jocher --- ultralytics/engine/validator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/validator.py b/ultralytics/engine/validator.py index 160a549d..5e0f0988 100644 --- a/ultralytics/engine/validator.py +++ b/ultralytics/engine/validator.py @@ -111,7 +111,7 @@ class BaseValidator: self.device = trainer.device self.data = trainer.data # force FP16 val during training - self.args.half = self.device.type != "cpu" and self.args.amp + self.args.half = self.device.type != "cpu" and trainer.amp model = trainer.ema.ema or trainer.model model = model.half() if self.args.half else model.float() # self.model = model