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