NVIDIA / NeMo-Run

A tool to configure, launch and manage your machine learning experiments.
Apache License 2.0
79 stars 20 forks source link

Add slurm_template parameter to SlurmExecutor (and use it in SlurmBatchRequest) #15

Open Kipok opened 3 months ago

Kipok commented 3 months ago

We have a need to customize the current slurm template used here https://github.com/NVIDIA/NeMo-Run/blob/main/src/nemo_run/core/execution/templates/slurm.sh.j2. I imagine there might be more use-cases where customization is necessary, so it might be a good idea to accept custom template as a parameter (with existing one as default)