From a7edd418b1c6c2c7cb2c474874049381be0d908f Mon Sep 17 00:00:00 2001
From: Less Wright <lessw@etrillium.com>
Date: Mon, 10 May 2021 22:24:18 -0700
Subject: [PATCH] fix context mgr for closure and grad

---
 ranger21/ranger21.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ranger21/ranger21.py b/ranger21/ranger21.py
index ab29073..1a529b6 100644
--- a/ranger21/ranger21.py
+++ b/ranger21/ranger21.py
@@ -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