mirrors
/
dinov2
mirror of https://github.com/facebookresearch/dinov2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix mem_per_gpu

pull/422/head
Etienne Guevel 2024-05-15 16:52:13 +02:00
parent 874ac5192e
commit 252ca3b3ab
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dinov2/utils/cluster.py
View File

@ -76,8 +76,8 @@ def get_slurm_executor_parameters(
) -> Dict[str, Any]:
# create default parameters
params = {
"mem_gb": 0, # Requests all memory on a node, see https://slurm.schedmd.com/sbatch.html
"mem_per_gb": mem_per_gpu,
#"mem_gb": 0, # Requests all memory on a node, see https://slurm.schedmd.com/sbatch.html
"slurm_mem_per_gpu": mem_per_gpu,
"gpus_per_node": num_gpus_per_node,
"tasks_per_node": num_gpus_per_node, # one task per GPU
"cpus_per_task": 10,