Small hotfix of DDP doc

mala-project · May 25, 2024 · f7abbfc · f7abbfc
1 parent 7abd9d7
commit f7abbfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/advanced_usage/trainingmodel.rst b/docs/source/advanced_usage/trainingmodel.rst
@@ -272,7 +272,7 @@ following setup was confirmed to work on an HPC cluster using the
         echo "MASTER_ADDR="$MASTER_ADDR
 
         # Run using srun.
-        srun -u bash -c '
+        srun -N NUMBER_OF_NODES -u bash -c '
         # Export additional per process variables
         export RANK=$SLURM_PROCID
         export LOCAL_RANK=$SLURM_LOCALID