configs/sweep/example_submitit.yaml

# @package _global_
defaults:
  - basic
  - override /hydra/launcher: submitit_local

hydra:
  sweeper:
    params:
      preset: graph_dta,deep_dta
      experiment: chembl_random
  launcher:
    timeout_min: 60
    cpus_per_task: 1
    gpus_per_node: 1
    tasks_per_node: 4
    mem_gb: 32
    nodes: 1