#!/bin/bash
# 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