mnist.sh

#!/bin/bash

#SBATCH --job-name=PermMNIST        # Job name
#SBATCH --output=./slurm_out/output_%j.txt     # Standard output and error log (%j will be replaced by job ID)
#SBATCH --mail-type=begin,end,fail       # Notifications for job done or failed
#SBATCH --mail-user=xgu3km@virginia.edu  # Where to send mail
#SBATCH --ntasks=1                 # Number of tasks (usually set to 1 for single-threaded jobs)
#SBATCH --cpus-per-task=20          # Number of CPU cores per task
#SBATCH --gpus-per-node=1         # Number of GPU per node
#SBATCH --mem=128G                  # Total memory for job
#SBATCH --constraint=a100_80gb     # GPU model
#SBATCH --time=4-00:00:00            # Time limit hrs:min:sec
#SBATCH --partition=gpu        # Partition (queue) to submit to

# python perm_mnist_main.py --prefix=final --suffix=ffnn_no_memory -v
python perm_mnist_main.py --prefix=final --suffix=tf_sequential_softmax_no_mask -v --model=transformer -lr=0.0005