configurations/cluster/mit_supercloud.yaml

defaults:
  - base_slurm
  - _self_
is_compute_node_offline: True  # many slurm systems only allows internet on login node, not compute node

launch_template: |
  #!/bin/bash

  #SBATCH -J {name}
  #SBATCH -o {log_dir}/out_%j.out
  #SBATCH -e {log_dir}/error_%j.err
  #SBATCH --mail-user={email}
  #SBATCH --mail-type=FAIL
  #SBATCH --gres=gpu:volta:{num_gpus}
  #SBATCH --cpus-per-task={num_cpus}
  #SBATCH --mem={memory}
  #SBATCH --time={time}

  cd {project_root}
  module load anaconda/2023a
  
  python -m main {python_args}