fix context mgr for closure and grad

2026-09-09 11:15:43 +08:00 · 2021-05-10 22:24:18 -07:00
parent de331ad196
commit a7edd418b1
1 changed files with 1 additions and 1 deletions
@@ -512,7 +512,7 @@ class Ranger21(TO.Optimizer):

        loss = None
        if closure is not None and isinstance(closure, collections.Callable):
-            with torch.grad():
+            with torch.enable_grad():
                loss = closure()

        param_size = 0