Merge pull request #1083 from duncanspani/develop

Add workload var for added CLI args to py-nemo-2 application.

Author: douglasjacobsen

var/ramble/repos/builtin/applications/py-nemo-2/application.py

@@ -40,7 +40,7 @@ class PyNemo2(BasePyNemo):
         "pretraining_exec",
         'bash -c "cd /opt/NeMo; git rev-parse HEAD; '
         "{custom_injected_string}; "
-        'python3 -u {experiment_run_dir}/{nemo_config_name};"',
+        'python3 -u {experiment_run_dir}/{nemo_config_name} {nemo_cli_args};"',
         use_mpi=True,
     )
 
@@ -68,6 +68,13 @@ class PyNemo2(BasePyNemo):
         workload_group="pretraining",
     )
 
+    workload_variable(
+        "nemo_cli_args",
+        default="",
+        description="CLI args to pass to append to the Python Nemo command.",
+        workload_group="pretraining",
+    )
+
     workload_variable(
         "results_mount",
         default="{experiment_run_dir}:{experiment_run_dir}",