From 9c4f9205f34c44cab56e6dd1ab13c4dc8e84cc94 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Mon, 9 Mar 2020 07:18:35 +0000
Subject: [PATCH] fit for yolov3

---
 yolov3.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yolov3.py b/yolov3.py
index dbebf34..d314028 100644
--- a/yolov3.py
+++ b/yolov3.py
@@ -470,7 +470,7 @@ def run(model, loader, mode='train'):
     start = time.time()
 
     for idx, batch in enumerate(loader()):
-        outputs, losses = getattr(model, mode)(
+        losses, _ = getattr(model, mode)(
             batch[0], batch[1], device='gpu', device_ids=device_ids)
 
         total_loss += np.sum(losses)
-- 
GitLab