Merge pull request #184 from inferno-pytorch/fix_frequency

Handle value 'never' in train_utils.Frequency
inferno-pytorch · Jul 8, 2019 · e896cf5 · e896cf5
2 parents 80396d1 + 758925d
commit e896cf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inferno/utils/train_utils.py b/inferno/utils/train_utils.py
@@ -109,7 +109,7 @@ def value(self):
     @value.setter
     def value(self, value):
         # If value is not being set, make sure the frequency never matches muhahaha
-        if value is None:
+        if value is None or value == 'never':
             value = np.inf
         self.assert_value_consistent(value)
         self._value = value