Update README.md

This commit is contained in:
Srinivasan Iyer 2025-02-13 11:57:49 -08:00 committed by GitHub
parent 85c2f28f26
commit 67b6c3b3da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -63,11 +63,11 @@ Now launch a debug job to check if everything works. **The provided configuratio
```bash
# stool stands for SLURM tool !
python -m bytelatent.stool script=bytelatent.train config=apps/bytelatent/configs/debug.yaml nodes=1 partition=<partition>
python -m bytelatent.stool script=bytelatent.train config=bytelatent/configs/debug.yaml nodes=1 partition=<partition>
# if you want to launch locally you can use torchrun
torchrun --nproc-per-node 8 -m bytelatent.train config=apps/bytelatent/configs/debug.yaml
torchrun --nproc-per-node 8 -m bytelatent.train config=bytelatent/configs/debug.yaml
# or you can also launch on 1 GPU
python -m bytelatent.train config=apps/bytelatent/configs/debug.yaml
python -m bytelatent.train config=bytelatent/configs/debug.yaml
```
When using `stool`, if a job crashes, it can be relaunched using sbatch: