From bec94be01a4a0f5a0ae63c00de4ccf1203afc17d Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 22 Nov 2018 14:33:01 +0100
Subject: [PATCH] updates

---
 models.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/models.py b/models.py
index e851d29c..1b296061 100755
--- a/models.py
+++ b/models.py
@@ -254,8 +254,11 @@ class Darknet(nn.Module):
                 output.append(x)
             layer_outputs.append(x)
 
+        self.losses['nT'] /= 3
+        self.losses['TP'] = 0
+        self.losses['FP'] = 0
+        self.losses['FN'] = 0
         if is_training and requestPrecision:
-            self.losses['nT'] /= 3
             self.losses['TC'] /= 3  # target category
             metrics = torch.zeros(3, len(self.losses['FPe']))  # TP, FP, FN