Update docs/examples/data/torchvision/job.sh

Co-authored-by: Fabrice Normandin <fabrice.normandin@gmail.com>
mila-iqia · Sep 21, 2023 · c1b4fbb · c1b4fbb
1 parent c3846de
commit c1b4fbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/examples/data/torchvision/job.sh b/docs/examples/data/torchvision/job.sh
@@ -39,7 +39,7 @@ mkdir -p "$SLURM_TMPDIR/data"
 # Copy the dataset to $SLURM_TMPDIR so it is close to the GPUs for
 # faster training
 srun --ntasks=$SLURM_JOB_NUM_NODES --ntasks-per-node=1 \
-	time -p bash data.py "/network/datasets/inat" ${_DATA_PREP_WORKERS}
+	time -p python data.py "/network/datasets/inat" ${_DATA_PREP_WORKERS}
 
 
 # Fixes issues with MIG-ed GPUs with versions of PyTorch < 2.0