diff --git a/yolov3.py b/yolov3.py
index dbebf34243419cefef85b636c04e971428045bb5..d31402800e57502685191d9c3f5c0a1a3aea1c74 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)