updates

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