| #SBATCH -J h2h | |
| #SBATCH -p gpu-h100 | |
| #SBATCH --gpus=h100:1 | |
| #SBATCH -c 4 | |
| #SBATCH --mem=32g | |
| #SBATCH -t 3:00:00 | |
| #SBATCH -o logs/h2h-%j.out | |
| source /etc/profile | |
| SCR=/scratch/zt1/project/msml612/user/$USER | |
| module load python/3.10.10 | |
| export PYTHONPATH= | |
| cd $SCR/pcmd/src | |
| for D in 12 16; do | |
| $SCR/hf_env/bin/python -u beat_baselines.py --runs $SCR/pcmd/runs --digits $D \ | |
| --n_eval 400 --eval_bs 100 --out $SCR/pcmd/figures/beat_d$D.json | |
| done | |