used only cpus per task in base config (#11)

zaccharieramzi · Aug 18, 2022 · 24d6cd9 · 24d6cd9
1 parent 32da82e
commit 24d6cd9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -42,7 +42,7 @@ The parameters you can override are defined in the `hydra-submitit-launcher` plu
 
 For example, if you want to use the gpu_p2 partition, you would need to do:
 ```
-hydra-submitit-launch my_app.py dev hydra.launcher.setup=null hydra.launcher.partition=gpu_p2
+hydra-submitit-launch my_app.py dev hydra.launcher.setup=null hydra.launcher.partition=gpu_p2 hydra.launcher.cpus_per_task=3
 ```
 
 In order to change the timeout on the SLURM job to for example 10 hours, you would need to do:

diff --git a/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml b/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml
@@ -6,8 +6,7 @@ gpus_per_node: 1
 tasks_per_node: 1
 gres: "gpu:${hydra.launcher.gpus_per_node}"
 qos: ${qos_from_hours:${hours}}
-cpus_per_gpu: 10
-cpus_per_task: ${cpu_from_gpu:${hydra.launcher.gpus_per_node},${hydra.launcher.cpus_per_gpu}}
+cpus_per_task: ${cpu_from_gpu:${hydra.launcher.gpus_per_node},10}
 gpus_per_task: ${hydra.launcher.gpus_per_node}
 additional_parameters:
   account: ${oc.env:IDRPROJ}@gpu