Files
WCL/run_drl_200ep.sh

18 lines
547 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 1 2 3 4 5 6; do
# Skip U=4 (already have 200ep run in results_drl_long)
if [ $U -eq 4 ]; then continue; fi
echo ">>> DRL 200ep U=$U"
$PY Code/drl_mask_policy.py \
--mode drl --embed-file $EMB \
--users $U --mux-factor 4 \
--epochs 200 --steps-per-epoch 200 \
--save-dir results_sweeps/drl_U${U}_200ep \
2>&1 | tail -2
done
echo "=== DRL 200ep sweep done ==="