Update simclr.py

sthalles · Apr 18, 2020 · c057471 · c057471
1 parent df7500e
commit c057471
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/simclr.py b/simclr.py
@@ -138,13 +138,14 @@ def _validate(self, model, valid_loader):
             model.eval()
 
             valid_loss = 0.0
-            for counter, ((xis, xjs), _) in enumerate(valid_loader):
+            counter = 0
+            for (xis, xjs), _ in valid_loader:
                 xis = xis.to(self.device)
                 xjs = xjs.to(self.device)
 
                 loss = self._step(model, xis, xjs, counter)
                 valid_loss += loss.item()
-
+                counter += 1
             valid_loss /= counter
         model.train()
         return valid_loss