From 2daa412575dd81615042b918d02964e1971a605f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Jun 2020 16:28:25 -0700 Subject: [PATCH] train with multi-gpu half test bug fix #99 --- test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test.py b/test.py index df8ed67..da803b7 100644 --- a/test.py +++ b/test.py @@ -46,7 +46,8 @@ def test(data, else: # called by train.py training = True device = next(model.parameters()).device # get model device - half = device.type != 'cpu' and torch.cuda.device_count() == 1 # half precision only supported on single-GPU + # half disabled https://github.com/ultralytics/yolov5/issues/99 + half = False # device.type != 'cpu' and torch.cuda.device_count() == 1 if half: model.half() # to FP16