defaults:
  - base_slurm

params:
  env_name: dfot
  partition: vision-sitzmann
  qos: vision-sitzmann-main
  num_gpus: 8
  memory: 512G
  num_cpus: 128
  time: "72:0:0"

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:{num_gpus}
  #SBATCH --cpus-per-task={num_cpus}
  #SBATCH --mem={memory}
  #SBATCH --time={time}
  #SBATCH --partition={partition}
  #SBATCH --qos={qos}
  source /data/scratch/kiwhan/.zshrc
  conda activate {env_name}
  cd {project_root}
  python -m main {python_args}