fixes

awaelchli · awaelchli · commit 55ce21b7c5c4 · 2022-09-15T15:38:33.000+02:00
diff --git a/src/pytorch_lightning/lite/wrappers.py b/src/pytorch_lightning/lite/wrappers.py
@@ -62,9 +62,8 @@ def step(self, closure: Optional[Callable] = None, module: Optional["_LiteModule
         closure = closure or _do_nothing_closure
         return self._strategy.optimizer_step(
             self.optimizer,
-            opt_idx=0,
+            model=(module if module is not None else getattr(self._strategy, "model", None)),
             closure=closure,
-            model=(module if module is not None else getattr(self._strategy, "model", None))
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -62,9 +62,8 @@ def step(self, closure: Optional[Callable] = None, module: Optional["_LiteModule`
`62`	`62`	`closure = closure or _do_nothing_closure`
`63`	`63`	`return self._strategy.optimizer_step(`
`64`	`64`	`self.optimizer,`
`65`		`- opt_idx=0,`
	`65`	`+ model=(module if module is not None else getattr(self._strategy, "model", None)),`
`66`	`66`	`closure=closure,`
`67`		`- model=(module if module is not None else getattr(self._strategy, "model", None))`
`68`	`67`	`)`
`69`	`68`
`70`	`69`