Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jun 16, 2020

Commit

72d0614

•

1 Parent(s): bafbc65

Multi-GPU disabled in test.py

Files changed (1) hide show

test.py CHANGED Viewed

@@ -1,7 +1,6 @@
 import argparse
 import json
-import yaml
 from torch.utils.data import DataLoader
 from utils.datasets import *
@@ -40,8 +39,9 @@ def test(data,
         if half:
             model.half()  # to FP16
-        if device.type != 'cpu' and torch.cuda.device_count() > 1:
-            model = nn.DataParallel(model)
     else:  # called by train.py
         training = True

 import argparse
 import json
 from torch.utils.data import DataLoader
 from utils.datasets import *
         if half:
             model.half()  # to FP16
+        # Multi-GPU disabled, incompatible with .half()
+        # if device.type != 'cpu' and torch.cuda.device_count() > 1:
+        #     model = nn.DataParallel(model)
     else:  # called by train.py
         training = True