Skip to content

Commit

Permalink
[format] applied code formatting on changed files in pull request 3025 (
Browse files Browse the repository at this point in the history
hpcaitech#3026)

Co-authored-by: github-actions <[email protected]>
  • Loading branch information
github-actions[bot] and github-actions authored Mar 7, 2023
1 parent cd2b0ea commit e86d9bb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions applications/ChatGPT/examples/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@ Use these code to train your reward model.

```shell
# Naive reward model training
python train_reward_model.py --pretrain <your model path> --model <your model type> --strategy naive
python train_reward_model.py --pretrain <your model path> --model <your model type> --strategy naive
# use colossalai_zero2
torchrun --standalone --nproc_per_node=2 train_reward_model.py --pretrain <your model path> --model <your model type> --strategy colossalai_zero2
torchrun --standalone --nproc_per_node=2 train_reward_model.py --pretrain <your model path> --model <your model type> --strategy colossalai_zero2
```

## Train with dummy prompt data (Stage 3)
Expand Down

0 comments on commit e86d9bb

Please sign in to comment.