tweak MLproject layout

m2lines · Aug 31, 2023 · bcca14f · bcca14f
1 parent 6cc3051
commit bcca14f
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/MLproject b/MLproject
@@ -36,7 +36,17 @@ entry_points:
       submodel : {type: string, default : transform3}
       features_transform_cls_name : {type : string, default : None}
       targets_transform_cls_name : {type : string, default : None}
-    #command: "python src/gz21_ocean_momentum/trainScript.py {exp_id} {run_id} --batchsize {batchsize} --learning_rate {learning_rate} --n_epochs {n_epochs} --train_split {train_split} --test_split {test_split} --time_indices {time_indices} --printevery {print_every} --weight_decay {weight_decay} --model_module_name {model_module_name} --model_cls_name {model_cls_name} --loss_cls_name {loss_cls_name}
-    command: "python src/gz21_ocean_momentum/trainScript.py --forcing-data-path {forcing_data_path} --batchsize {batchsize} --learning_rate {learning_rate} --n_epochs {n_epochs} --train_split {train_split} --test_split {test_split} --time_indices {time_indices} --printevery {print_every} --weight_decay {weight_decay} --model_module_name {model_module_name} --model_cls_name {model_cls_name} --loss_cls_name {loss_cls_name}
-    --transformation_cls_name {transformation_cls_name} --submodel {submodel} --features_transform_cls_name {features_transform_cls_name} --targets_transform_cls_name {targets_transform_cls_name}"
-
+        #{exp_id} {run_id}
+    command: "python src/gz21_ocean_momentum/trainScript.py
+        --forcing-data-path {forcing_data_path}
+        --batchsize {batchsize} --learning_rate {learning_rate}
+        --n_epochs {n_epochs} --train_split {train_split}
+        --test_split {test_split} --time_indices {time_indices}
+        --printevery {print_every} --weight_decay {weight_decay}
+        --model_module_name {model_module_name}
+        --model_cls_name {model_cls_name}
+        --loss_cls_name {loss_cls_name}
+        --transformation_cls_name {transformation_cls_name}
+        --submodel {submodel}
+        --features_transform_cls_name {features_transform_cls_name}
+        --targets_transform_cls_name {targets_transform_cls_name}"