uer commited on
Commit
5b004cb
1 Parent(s): fc2f739

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -120,7 +120,7 @@ python3 preprocess.py --corpus_path corpora/cluecorpussmall.txt \
120
  ```
121
  python3 pretrain.py --dataset_path cluecorpussmall_seq128_dataset.pt \
122
  --vocab_path models/google_zh_vocab.txt \
123
- --config_path models/bert_medium_config.json \
124
  --output_model_path models/cluecorpussmall_roberta_medium_seq128_model.bin \
125
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
126
  --total_steps 1000000 --save_checkpoint_steps 100000 --report_steps 50000 \
@@ -142,7 +142,7 @@ python3 preprocess.py --corpus_path corpora/cluecorpussmall.txt \
142
  python3 pretrain.py --dataset_path cluecorpussmall_seq512_dataset.pt \
143
  --pretrained_model_path models/cluecorpussmall_roberta_medium_seq128_model.bin-1000000 \
144
  --vocab_path models/google_zh_vocab.txt \
145
- --config_path models/bert_medium_config.json \
146
  --output_model_path models/cluecorpussmall_roberta_medium_seq512_model.bin \
147
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
148
  --total_steps 250000 --save_checkpoint_steps 50000 --report_steps 10000 \
120
  ```
121
  python3 pretrain.py --dataset_path cluecorpussmall_seq128_dataset.pt \
122
  --vocab_path models/google_zh_vocab.txt \
123
+ --config_path models/bert/medium_config.json \
124
  --output_model_path models/cluecorpussmall_roberta_medium_seq128_model.bin \
125
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
126
  --total_steps 1000000 --save_checkpoint_steps 100000 --report_steps 50000 \
142
  python3 pretrain.py --dataset_path cluecorpussmall_seq512_dataset.pt \
143
  --pretrained_model_path models/cluecorpussmall_roberta_medium_seq128_model.bin-1000000 \
144
  --vocab_path models/google_zh_vocab.txt \
145
+ --config_path models/bert/medium_config.json \
146
  --output_model_path models/cluecorpussmall_roberta_medium_seq512_model.bin \
147
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
148
  --total_steps 250000 --save_checkpoint_steps 50000 --report_steps 10000 \