mirror of
https://github.com/karpathy/llm.c.git
synced 2026-07-28 20:35:09 -04:00
also updated pyrun124M.sh
This commit is contained in:
parent
03c2e85601
commit
ea88a343b5
1 changed files with 1 additions and 1 deletions
|
|
@ -3,7 +3,7 @@
|
|||
# if you wish to train on just a single GPU, simply skip the torchrun part, i.e.
|
||||
# python train_gpt2.py ... (all the other arguments the same)
|
||||
torchrun --standalone --nproc_per_node=4 train_gpt2.py \
|
||||
--input_bin dev/data/fineweb10B/fineweb_train_000001.bin \
|
||||
--input_bin "dev/data/fineweb10B/fineweb_train_*.bin" \
|
||||
--output_dir pylog124M \
|
||||
--write_tensors 0 \
|
||||
--model d12 \
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue