#!/bin/bash
set -e
cd "/Users/kyo/Documents/AY/논문/5. WCL-DRL"
EMB="bert_agnews_8000.pt"
PY="/opt/anaconda3/bin/python"

# Ablation variants at U=4, 100ep, 3 seeds
for S in 0 42 123; do
    echo ">>> DRL beta=0 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 \
        --beta 0.0 --seed $S \
        --save-dir results_sweeps/drl_beta0_100ep_s${S} \
        2>&1 | tail -2

    echo ">>> DRL r=16 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 \
        --rank 16 --seed $S \
        --save-dir results_sweeps/drl_r16_100ep_s${S} \
        2>&1 | tail -2
done

# U-sweep at U={1,2,3,5,6} for throughput, 100ep, 3 seeds
for S in 0 42 123; do
    for U in 1 2 3 5 6; do
        echo ">>> Joint U=$U 100ep seed=$S"
        $PY Code/drl_mask_policy.py \
            --mode joint --embed-file $EMB \
            --users $U --mux-factor 4 \
            --epochs 100 --steps-per-epoch 200 \
            --seed $S \
            --save-dir results_sweeps/joint_U${U}_100ep_s${S} \
            2>&1 | tail -2
        echo ">>> DRL U=$U 100ep seed=$S"
        $PY Code/drl_mask_policy.py \
            --mode drl --embed-file $EMB \
            --users $U --mux-factor 4 \
            --epochs 100 --steps-per-epoch 200 \
            --seed $S \
            --save-dir results_sweeps/drl_U${U}_100ep_s${S} \
            2>&1 | tail -2
    done
done

echo "=== all multi-seed 100ep runs done ==="