From 0bc111f0b423f050db8790caa6f2306bcf0f2567 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 16 Nov 2018 22:35:44 +0100 Subject: [PATCH] updates --- models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/models.py b/models.py index 4eb65f37..418a53db 100755 --- a/models.py +++ b/models.py @@ -141,7 +141,6 @@ class YOLOLayer(nn.Module): # Training if targets is not None: - self.batch_count += 1 MSELoss = nn.MSELoss() BCEWithLogitsLoss = nn.BCEWithLogitsLoss() CrossEntropyLoss = nn.CrossEntropyLoss()