update runners

Author: wlnc

cv.py

@@ -28,15 +28,18 @@
 # ------------------------------------------------------------------ #
 
 
-def _add_shared_args(p: argparse.ArgumentParser) -> None:
+def _add_shared_args(
+    p: argparse.ArgumentParser, *, include_portfolio: bool = True
+) -> None:
     p.add_argument("name", help="Experiment name (used for output file names)")
-    p.add_argument(
-        "-p",
-        "--portfolio",
-        nargs="+",
-        default=["SPSO", "IPSO", "SPSOL"],
-        help="Sub-optimizer names from the portfolio",
-    )
+    if include_portfolio:
+        p.add_argument(
+            "-p",
+            "--portfolio",
+            nargs="+",
+            default=["SPSO", "IPSO", "SPSOL"],
+            help="Sub-optimizer names from the portfolio",
+        )
     p.add_argument(
         "--fe-multiplier",
         type=int,
@@ -121,7 +124,7 @@ def _parse_args() -> argparse.Namespace:
         help="Custom RL-DAS: single-dimension, pure-PyTorch PPO",
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
-    _add_shared_args(rl)
+    _add_shared_args(rl, include_portfolio=False)
     rl.add_argument(
         "--dim", type=int, default=10, help="Problem dimension (agent is dim-specific)"
     )

rl_das_study.slurm

@@ -10,36 +10,28 @@
 #SBATCH -A plgrldas2026-gpu-a100
 #SBATCH --array=0-7
 
-# Args: SEED [PORTFOLIO...]
+# Args: SEED
 SEED=${1:-42}
 
-if [ "$#" -lt 2 ]; then
-    PORTFOLIO=('SPSO' 'IPSO' 'SPSOL')
-else
-    PORTFOLIO=("${@:2}")
-fi
-
-PORTFOLIO_STR=$(IFS="_"; echo "${PORTFOLIO[*]}")
-
 ENV_PATH="$SCRATCH/DynamicAlgorithmSelection2/.venv/bin/activate"
 source "$ENV_PATH"
 mkdir -p logs
 
 DIMS=(2 3 5 10)
 
-echo "Array job $SLURM_ARRAY_TASK_ID | SEED=$SEED | PORTFOLIO=${PORTFOLIO[*]}"
+echo "Array job $SLURM_ARRAY_TASK_ID | SEED=$SEED"
 
 # 0-3: CV-LOIO per dimension
 if [[ $SLURM_ARRAY_TASK_ID -ge 0 && $SLURM_ARRAY_TASK_ID -le 3 ]]; then
     DIM=${DIMS[$SLURM_ARRAY_TASK_ID]}
     echo "RL-DAS | CV-LOIO | dim=$DIM"
-    python cv.py rl-das ${PORTFOLIO_STR}_RLDAS_LOIO_DIM${DIM}_SEED${SEED} \
-        -p "${PORTFOLIO[@]}" --dim $DIM --cv-mode LOIO --n-epochs 500 --seed $SEED
+    python cv.py rl-das NL_SHADE_RSP_MADDE_JDE21_RLDAS_LOIO_DIM${DIM}_SEED${SEED} \
+        --dim $DIM --cv-mode LOIO --n-epochs 500 --seed $SEED
 
 # 4-7: CV-LOPO per dimension
 elif [[ $SLURM_ARRAY_TASK_ID -ge 4 && $SLURM_ARRAY_TASK_ID -le 7 ]]; then
     DIM=${DIMS[$((SLURM_ARRAY_TASK_ID - 4))]}
     echo "RL-DAS | CV-LOPO | dim=$DIM"
-    python cv.py rl-das ${PORTFOLIO_STR}_RLDAS_LOPO_DIM${DIM}_SEED${SEED} \
-        -p "${PORTFOLIO[@]}" --dim $DIM --cv-mode LOPO --n-epochs 500 --seed $SEED
+    python cv.py rl-das NL_SHADE_RSP_MADDE_JDE21_RLDAS_LOPO_DIM${DIM}_SEED${SEED} \
+        --dim $DIM --cv-mode LOPO --n-epochs 500 --seed $SEED
 fi

runner.sh

@@ -9,16 +9,17 @@ PORTFOLIOS=(
 echo "Starting job submissions..."
 
 for SEED in "${SEEDS[@]}"; do
+
+    echo "Submitting RL-DAS study | SEED=$SEED"
+    sbatch rl_das_study.slurm $SEED
+    sleep 1
+
     for PORTFOLIO in "${PORTFOLIOS[@]}"; do
 
         echo "Submitting PPO study | SEED=$SEED | PORTFOLIO=$PORTFOLIO"
         sbatch ppo_study.slurm $SEED $PORTFOLIO
         sleep 1
 
-        echo "Submitting RL-DAS study | SEED=$SEED | PORTFOLIO=$PORTFOLIO"
-        sbatch rl_das_study.slurm $SEED $PORTFOLIO
-        sleep 1
-
         echo "Submitting Exp-DAS study | SEED=$SEED | PORTFOLIO=$PORTFOLIO"
         sbatch exp_das_study.slurm $SEED $PORTFOLIO
         sleep 1

train.py

@@ -41,15 +41,18 @@
 # ------------------------------------------------------------------ #
 
 
-def _add_shared_args(p: argparse.ArgumentParser) -> None:
+def _add_shared_args(
+    p: argparse.ArgumentParser, *, include_portfolio: bool = True
+) -> None:
     p.add_argument("name", help="Experiment name (used for output file names)")
-    p.add_argument(
-        "-p",
-        "--portfolio",
-        nargs="+",
-        default=["SPSO", "IPSO", "SPSOL"],
-        help="Sub-optimizer names from the portfolio",
-    )
+    if include_portfolio:
+        p.add_argument(
+            "-p",
+            "--portfolio",
+            nargs="+",
+            default=["SPSO", "IPSO", "SPSOL"],
+            help="Sub-optimizer names from the portfolio",
+        )
     p.add_argument(
         "--mode",
         choices=["easy", "hard", "random"],
@@ -131,7 +134,7 @@ def _parse_args() -> argparse.Namespace:
         help="Custom RL-DAS: single-dimension, pure-PyTorch PPO",
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
-    _add_shared_args(rl)
+    _add_shared_args(rl, include_portfolio=False)
     rl.add_argument(
         "--dim", type=int, default=10, help="Problem dimension (agent is dim-specific)"
     )