switch order of args (NVIDIA#5549)

Signed-off-by: ericharper <[email protected]> Signed-off-by: ericharper <[email protected]> Co-authored-by: Sandeep Subramanian <[email protected]>
titu1994 · Mar 24, 2023 · 3eea482 · 3eea482
1 parent 51d4c38
commit 3eea482
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/parts/nlp_overrides.py b/nemo/collections/nlp/parts/nlp_overrides.py
@@ -590,8 +590,8 @@ def __init__(
 
     def optimizer_step(
         self,
-        model: Union["pl.LightningModule", torch.nn.Module],
         optimizer: torch.optim.Optimizer,
+        model: Union["pl.LightningModule", torch.nn.Module],
         optimizer_idx: int,
         closure: Callable[[], Any],
         **kwargs: Any,