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]> Signed-off-by: andrusenkoau <[email protected]>
andrusenkoau · Jan 5, 2023 · ca5cdc5 · ca5cdc5
1 parent ccbd4dd
commit ca5cdc5
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,