Skip to content

Commit 5ce7362

Browse files
Separiusdanielkorzekwa
authored andcommitted
replace submit_job with srun
Signed-off-by: Daniel Korzekwa <dkorzekwa@nvidia.com>
1 parent dd010f1 commit 5ce7362

File tree

1 file changed

+5
-6
lines changed
  • examples/puzzletron/automodel_distillation

1 file changed

+5
-6
lines changed

examples/puzzletron/automodel_distillation/README.md

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -88,12 +88,11 @@ torchrun --nproc_per_node=2 \
8888
Below is an example job setup: NeMo AutoModel container, clone AutoModel main, install it and upgrade Transformers, then run KD from a directory that contains your config and run script (e.g. a copy of this example or the RealAnyModel layout).
8989

9090
```bash
91-
# Submit interactive job (example with your cluster’s submit_job)
92-
submit_job --partition interactive --time 2.0 \
93-
--image nvcr.io/nvidia/nemo-automodel:25.11.00 \
94-
--mounts "/path/to/AutoModel/:/opt/Automodel/,/lustre:/lustre" \
95-
--interactive --gpu 2 --skip_image_check --email_mode=never \
96-
--command='bash'
91+
# Submit interactive job
92+
srun --partition=interactive --time=2:00:00 --gres=gpu:2 \
93+
--container-image=nvcr.io/nvidia/nemo-automodel:25.11.00 \
94+
--container-mounts="/path/to/AutoModel/:/opt/Automodel/" \
95+
--pty bash
9796

9897
# Inside the container
9998
source /opt/venv/bin/activate

0 commit comments

Comments
 (0)