#!/bin/bash
set -e
cd "/Users/kyo/Documents/AY/논문/5. WCL-DRL"
EMB="bert_agnews_8000.pt"
PY="/opt/anaconda3/bin/python"
SEEDS="0 42 123 7 2025 2026"
for S in $SEEDS; do
    echo ">>> Joint U=4 100ep seed=$S"
    $PY Code/drl_mask_policy.py \
        --mode joint --embed-file $EMB \
        --users 4 --mux-factor 4 \
        --epochs 100 --steps-per-epoch 200 \
        --seed $S \
        --save-dir results_sweeps/joint_U4_100ep_s${S} \
        2>&1 | tail -2
    echo ">>> DRL U=4 100ep seed=$S"
    $PY Code/drl_mask_policy.py \
        --mode drl --embed-file $EMB \
        --users 4 --mux-factor 4 \
        --epochs 100 --steps-per-epoch 200 \
        --seed $S \
        --save-dir results_sweeps/drl_U4_100ep_s${S} \
        2>&1 | tail -2
done
echo "=== 6-seed 100ep sweep done ==="