train with multi-gpu half test bug fix #99

5 years ago · 2daa412575
parent d1e5716610
commit 2daa412575
1 changed files with 2 additions and 1 deletions
--- 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