From a4b8815ed934ce295b99c348b201f35e94ffd45f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 17 Jan 2020 17:58:37 -0800
Subject: [PATCH] updates

---
 test.py  | 4 ++--
 train.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/test.py b/test.py
index 68b636b1..7f809a5c 100644
--- a/test.py
+++ b/test.py
@@ -16,9 +16,9 @@ def test(cfg,
          conf_thres=0.001,
          iou_thres=0.5,  # for nms
          save_json=False,
+         single_cls=False,
          model=None,
-         dataloader=None,
-         single_cls=False):
+         dataloader=None):
     # Initialize/load model and set device
     if model is None:
         device = torch_utils.select_device(opt.device, batch_size=batch_size)
diff --git a/train.py b/train.py
index 158cbbef..859368af 100644
--- a/train.py
+++ b/train.py
@@ -319,8 +319,8 @@ def train():
                                       conf_thres=0.001 if final_epoch else 0.1,  # 0.1 for speed
                                       iou_thres=0.6 if final_epoch and is_coco else 0.5,
                                       save_json=final_epoch and is_coco,
-                                      dataloader=testloader,
-                                      single_cls=opt.single_cls)
+                                      single_cls=opt.single_cls,
+                                      dataloader=testloader)
 
         # Update scheduler
         scheduler.step()