#For a cluster using SLURM schedulers, the algorithm can be submitted using the following script.
#!/bin/bash
#NECESSARY JOB SPECIFICATIONS
#SBATCH --job-name=pauc #Set the job name
#SBATCH --time=5:00:00 #Set the wall clock limit to 4hrs
#SBATCH --ntasks-per-node=1
#SBATCH --nodes=16
#SBATCH --mem=16gb #Request 16GB Memory per node
#SBATCH --output=pauc.%j #stdout/err
#SBATCH --gres=gpu:1 #Request 1 GPU per node
#SBATCH --partition=gpu #Request the GPU partition/queue
export MASTER_PORT=12381
export WORLD_SIZE=16
echo "NODELIST="${SLURM_NODELIST}
master_addr=$(scontrol show hostnames "$SLURM_JOB_NODELIST" | head -n 1)
export MASTER_ADDR=$master_addr
echo "MASTER_ADDR="$MASTER_ADDR
#First Executable Line
# module load CUDA/11.3.1
module load Anaconda3/2021.11
source ~/.bashrc
conda activate YOUR-CONDA-ENVIROMENT
srun python3 main.py --OPAUC=1 --ce=0 --tau=1 --I=32 --lr=0.2 \
--B1=32 --B2=32 --gamma=0.1 --noise_std=0.2\
--data=chestmnist --pos_class=1 --class_index=4