From 2516e6c98f2535869cc9e07bbeca2545f99dcfbc Mon Sep 17 00:00:00 2001
From: Alessia Marcolini <amarcolini@fbk.eu>
Date: Thu, 16 Apr 2020 11:13:14 +0200
Subject: [PATCH] Fix validation loss calculation

---
 simclr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/simclr.py b/simclr.py
index 9a58f94a..0535bac5 100644
--- a/simclr.py
+++ b/simclr.py
@@ -145,6 +145,6 @@ def _validate(self, model, valid_loader):
                 loss = self._step(model, xis, xjs, counter)
                 valid_loss += loss.item()
 
-            valid_loss /= counter
+            valid_loss = valid_loss / (counter + 1)
         model.train()
         return valid_loss