configs/local/slurm_example.yaml

# @package _global_
defaults:
  - override /hydra/launcher: submitit_slurm

hydra:
  launcher:
    submitit_folder: ${hydra.sweep.dir}/.submitit
    name: "${hydra.job.name} ${eval:\"' '.join(${hydra.overrides.task}\")}"
    timeout_min: 6000
    cpus_per_task: 4
    gpus_per_task: 1
    gres: gpu:1
    partition: gpu3090
    qos: gpu3090
    additional_parameters:
      error: ${hydra.launcher.submitit_folder}/%j_%t.log
      output: ${hydra.launcher.submitit_folder}/%j_%t.log
    array_parallelism: 256

trainer:
  precision: bf16