#! /bin/bash
#SBATCH --nodes=1
#SBATCH --mem=64G
#SBATCH --qos=system
#SBATCH --job-name=pnpl
#SBATCH --time=11:59:00
#SBATCH --partition=short

source ~/.init_conda.sh
conda activate pnpl_base

export WANDB_CACHE_DIR=$DATA/wandb_cache

srun python $@
