Merge pull request evanmy#2 from varunj/patch-1

typo correction
yegmor · Apr 22, 2021 · b2fe7c4 · b2fe7c4
2 parents e6f67f4 + 68f26ff
commit b2fe7c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/functions/training_tools.py b/functions/training_tools.py
@@ -103,7 +103,7 @@ def padder(input, kernel_size):
 
 def normalize_dim1(x):
     '''
-    Ensure that dim1 sums up to zero for proper probabilistic interpretation
+    Ensure that dim1 sums up to one for proper probabilistic interpretation
     '''
     normalizer = torch.sum(x, dim=1, keepdim=True)
     return x/normalizer
@@ -122,4 +122,4 @@ def sequential(self):
             return next(self.iterator)
 
     def shuffle(self):
-        self.iterator = iter(sklearn.utils.shuffle(self.idx))
+        self.iterator = iter(sklearn.utils.shuffle(self.idx))