29 lines
912 B
Bash
Executable File
29 lines
912 B
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
cd "/Users/kyo/Documents/AY/논문/5. WCL-DRL"
|
|
EMB="bert_agnews_8000.pt"
|
|
PY="/opt/anaconda3/bin/python"
|
|
|
|
for U in 2 6; do
|
|
for S in 0 42 123; do
|
|
echo ">>> DRL beta=0 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 \
|
|
--beta 0.0 --seed $S \
|
|
--save-dir results_sweeps/drl_beta0_U${U}_100ep_s${S} \
|
|
2>&1 | tail -2
|
|
echo ">>> DRL r=16 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 \
|
|
--rank 16 --seed $S \
|
|
--save-dir results_sweeps/drl_r16_U${U}_100ep_s${S} \
|
|
2>&1 | tail -2
|
|
done
|
|
done
|
|
|
|
echo "=== U=2/U=6 ablation done ==="
|