Fix fastai#2526

jiapyliu · Mar 25, 2020 · 8d9d310 · 8d9d310
1 parent 9a32a5f
commit 8d9d310
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastai/callback.py b/fastai/callback.py
@@ -81,7 +81,7 @@ def lr(self, val:float)->None:
         self._lr = self.set_val('lr', listify(val, self._lr))
 
     @property
-    def mom(self)->float:return self._mom[-1]
+    def mom(self)->float:return self._mom[-1] if self._mom is not None else None
     @mom.setter
     def mom(self, val:float)->None:
         if 'momentum' in self.opt_keys: self.set_val('momentum', listify(val, self._mom))