Update README.md

README.md

@@ -44,7 +44,7 @@ When the parameter skip_special_tokens is False:
 
 ## Training data
 
-Contains 800,000 Chinese ancient poems collected by [chinese-poetry](https://github.com/chinese-poetry/chinese-poetry) and [Poetry](https://github.com/Werneror/Poetry) projects.
+Training data contains 800,000 Chinese ancient poems which are collected by [chinese-poetry](https://github.com/chinese-poetry/chinese-poetry) and [Poetry](https://github.com/Werneror/Poetry) projects.
 
 ## Training procedure
 
@@ -53,14 +53,14 @@ The model is pre-trained by [UER-py](https://github.com/dbiir/UER-py/) on [Tence
 ```
 python3 preprocess.py --corpus_path corpora/poem.txt \
 					  --vocab_path models/google_zh_vocab.txt \  
-					  --dataset_path poem.pt --processes_num 16 \
+					  --dataset_path poem_dataset.pt --processes_num 16 \
 					  --seq_length 128 --target lm 
 ```
 
 ```
-python3 pretrain.py --dataset_path poem.pt \
+python3 pretrain.py --dataset_path poem_dataset.pt \
 				    --vocab_path models/google_zh_vocab.txt \
-					--output_model_path models/poem_gpt_base_model.bin \  
+					--output_model_path models/poem_gpt2_base_model.bin \  
 					--config_path models/bert_base_config.json --learning_rate 5e-4 \
 					--tie_weight --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
 					--embedding word_pos --remove_embedding_layernorm \
@@ -72,7 +72,7 @@ python3 pretrain.py --dataset_path poem.pt \
 
 Finally, we convert the pre-trained model into Huggingface's format:
 ```
-python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path poem_gpt_base_model.bin \
+python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path poem_gpt2_base_model.bin-200000 \
 														--output_model_path pytorch_model.bin \
 														--layers_num 12
 ```