From d5db50df8ea3a5fab183821509dcfe1849b8c889 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 11 Apr 2019 18:26:52 +0200
Subject: [PATCH] updates

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

diff --git a/test.py b/test.py
index 25ce611e..5536eab0 100644
--- a/test.py
+++ b/test.py
@@ -171,7 +171,7 @@ def test(
         map = cocoEval.stats[1]  # update mAP to pycocotools mAP
 
     # Return results
-    return mp, mr, map, mf1, loss
+    return mp, mr, map, mf1, loss / len(dataloader)
 
 
 if __name__ == '__main__':