bug fixed

VCasecnikovs · Dec 4, 2020 · 0b47389 · 0b47389
1 parent d010272
commit 0b47389
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ranger.py b/ranger.py
@@ -88,8 +88,8 @@ def step(self, closure=None):
         # note - below is commented out b/c I have other work that passes back the loss as a float, and thus not a callable closure.
         # Uncomment if you need to use the actual closure...
 
-        # if closure is not None:
-        #loss = closure()
+        if closure is not None:
+            loss = closure()
 
         # Evaluate averages and grad, update param tensors
         for group in self.param_groups: