From 959c67b4eddc79d9cd35cb3fdee7cbab6a4010c8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 22 Nov 2018 14:48:57 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 1b296061..f45d65d2 100755 --- a/models.py +++ b/models.py @@ -273,9 +273,9 @@ class Darknet(nn.Module): self.losses['TP'] = metrics[0].sum() self.losses['FP'] = metrics[1].sum() self.losses['FN'] = metrics[2].sum() - self.losses['TC'] = 0 self.losses['metrics'] = metrics + self.losses['TC'] = 0 return sum(output) if is_training else torch.cat(output, 1)