- fixed bug in AutoLR: cooldown duration was basically ignored

inferno-pytorch · Feb 14, 2018 · a96dd32 · a96dd32
1 parent b0d03fd
commit a96dd32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inferno/trainers/callbacks/scheduling.py b/inferno/trainers/callbacks/scheduling.py
@@ -224,7 +224,7 @@ def out_of_patience(self):
     @property
     def in_cooldown(self):
         if self.cooldown_duration is not None:
-            return not self.patience.match(**self.duration_since_last_decay)
+            return not self.cooldown_duration.match(**self.duration_since_last_decay)
         else:
             return False