From 72d06147b20edfc1b3227c5443ee6bb79ac8ef2a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 16 Jun 2020 14:42:30 -0700
Subject: [PATCH] Multi-GPU disabled in test.py

---
 test.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test.py b/test.py
index 1f937a7..aa5cde3 100644
--- a/test.py
+++ b/test.py
@@ -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)
+        # 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