From f7abbfc9c88efc942b8a87020cde4c63d94210f7 Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Sat, 25 May 2024 20:16:48 +0200 Subject: [PATCH] Small hotfix of DDP doc --- docs/source/advanced_usage/trainingmodel.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/advanced_usage/trainingmodel.rst b/docs/source/advanced_usage/trainingmodel.rst index 4413ab078..52e50ec50 100644 --- 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