mirror of
https://github.com/wassname/Ranger21.git
synced 2026-09-09 11:15:43 +08:00
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user