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

Author: zaccharieramzi

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:

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