| # CPU-only: generate the hard Sudoku benchmark. No GPU needed, so this costs | |
| # almost nothing against the class GPU allocation. | |
| #SBATCH -J genhard | |
| #SBATCH -p standard | |
| #SBATCH -c 64 | |
| #SBATCH --mem=64g | |
| #SBATCH -t 3:00:00 | |
| #SBATCH -o logs/genhard-%j.out | |
| source /etc/profile | |
| SCR=/scratch/zt1/project/msml612/user/$USER | |
| module load python/3.10.10 | |
| export PYTHONPATH= | |
| cd $SCR/pcmd/src | |
| $SCR/hf_env/bin/python -u gen_hard_sudoku.py --n_train 100000 --n_test 1000 \ | |
| --workers 64 --out $SCR/pcmd/data/sudoku_hard | |