update readme

pull/2484/head
jiaruifang 2023-01-16 15:54:36 +08:00
parent 38424db6ff
commit 438ea608f3
1 changed files with 1 additions and 1 deletions

View File

@ -24,7 +24,7 @@ Use the following commands to execute training.
# export DATA=/path/to/small-gpt-dataset.json'
# run on a single node
colossalai run --nproc_per_node=<num_gpus> train_gpt.py --config configs/<config_file> --from_torch
colossalai run --nproc_per_node=<num_gpus> train_gpt.py --config configs/<config_file> --from_torch --use_dummy_dataset
# run on multiple nodes with slurm
colossalai run --nproc_per_node=<num_gpus> \