updates

2018-11-16 22:35:44 +01:00 · 2018-11-16 22:35:44 +01:00 · 0bc111f0b4
commit 0bc111f0b4
parent ed1067bfb5
1 changed files with 0 additions and 1 deletions
--- 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()