File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
var/ramble/repos/builtin/applications/py-nemo-2 Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -29,13 +29,13 @@ class PyNemo2(ExecutableApplication):
2929 tags ("ml-framework" , "machine-learning" )
3030
3131 # Add Nemo 2.0 config to archive.
32- archive_pattern ("{experiment_run_dir}/{nemo_2_config_name} " )
32+ archive_pattern ("{experiment_run_dir}/{nemo_config_name}/* " )
3333
3434 executable (
3535 "pretraining_exec" ,
3636 'bash -c "'
3737 "{custom_injected_string}; "
38- 'python3 -u {experiment_run_dir}/{nemo_2_config_name };"' ,
38+ 'python3 -u {experiment_run_dir}/{nemo_config_name };"' ,
3939 use_mpi = True ,
4040 )
4141
@@ -79,16 +79,16 @@ class PyNemo2(ExecutableApplication):
7979 )
8080
8181 workload_variable (
82- "nemo_2_config_dir_path " ,
82+ "nemo_config_dir_path " ,
8383 default = "" ,
8484 description = "Path to dir containing Nemo 2.0 python configs to be used." ,
8585 workload_group = "all_workloads" ,
8686 )
8787
8888 workload_variable (
89- "nemo_2_config_name " ,
89+ "nemo_config_name " ,
9090 default = "" ,
91- description = "Name of NeMo 2.0 config under {nemo_2_config_dir_path }." ,
91+ description = "Name of NeMo 2.0 config under {nemo_config_dir_path }." ,
9292 workload_group = "all_workloads" ,
9393 )
9494
You can’t perform that action at this time.
0 commit comments