remove unnecessary command-line arguments

Helmholtz-AI-Energy · Jan 14, 2025 · 3fcb08c · 3fcb08c
1 parent 51c5865
commit 3fcb08c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/scripts/experiments/generate_parallel_evaluation_from_breaking_iid_ckpt_job_scripts.py b/scripts/experiments/generate_parallel_evaluation_from_breaking_iid_ckpt_job_scripts.py
@@ -85,7 +85,7 @@ def generate_parallel_evaluation_from_breaking_iid_ckpt_job_scripts(
 mkdir -p "${{RESDIR}}"
 cd "${{RESDIR}}" || exit
 
-srun python -u ${{BASE_DIR}}/${{SCRIPT}} \\
+srun --mpi=pmix python -u ${{BASE_DIR}}/${{SCRIPT}} \\
     --n_samples {10**log_n_samples} \\
     --n_features {10**log_n_features} \\
     --n_classes {n_classes} \\
@@ -100,9 +100,7 @@ def generate_parallel_evaluation_from_breaking_iid_ckpt_job_scripts(
     --n_trees {n_nodes * n_trees} \\
     --random_state_model {model_seed} \\
     --output_dir ${{RESDIR}} \\
-    --output_label ${{SLURM_JOB_ID}} \\
-    --detailed_evaluation \\
-    --save_model
+    --output_label ${{SLURM_JOB_ID}}
                                 """
     output_path = output_path / f"nodes_{n_nodes}"
     os.makedirs(output_path, exist_ok=True)