Use collections.abc.Callable

Fixes #43
This commit is contained in:
Jack Kelly
2022-05-06 17:16:44 +01:00
committed by GitHub
parent 0a906ef9df
commit b507df6a4e
+1 -1
View File
@@ -575,7 +575,7 @@ class Ranger21(TO.Optimizer):
def step(self, closure=None):
loss = None
if closure is not None and isinstance(closure, collections.Callable):
if closure is not None and isinstance(closure, collections.abc.Callable):
with torch.enable_grad():
loss = closure()