updates

2019-04-11 18:26:52 +02:00
parent c9a55a269b
commit d5db50df8e
1 changed files with 1 additions and 1 deletions
@@ -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__':