Hugging Face's logo

uer
/
gpt2-chinese-poem

Text Generation
Transformers
PyTorch
TensorFlow
JAX
Chinese
gpt2
Inference Endpoints
text-generation-inference
Model card Files Community
22
uer commited on
Commit
46720b8
•
1 Parent(s): dfd153a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,7 +50,7 @@ The model is pre-trained by [UER-py](https://github.com/dbiir/UER-py/) on [Tence
50
 
51
  ```
52
  python3 preprocess.py --corpus_path corpora/poem.txt \
53
- --vocab_path models/google_zh_vocab.txt \
54
  --dataset_path poem_dataset.pt --processes_num 16 \
55
  --seq_length 128 --target lm
56
  ```
@@ -58,7 +58,7 @@ python3 preprocess.py --corpus_path corpora/poem.txt \
58
  ```
59
  python3 pretrain.py --dataset_path poem_dataset.pt \
60
  --vocab_path models/google_zh_vocab.txt \
61
- --output_model_path models/poem_gpt2_base_model.bin \
62
  --config_path models/bert_base_config.json \
63
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
64
  --total_steps 200000 --save_checkpoint_steps 50000 --report_steps 1000 \
 
50
 
51
  ```
52
  python3 preprocess.py --corpus_path corpora/poem.txt \
53
+ --vocab_path models/google_zh_vocab.txt \
54
  --dataset_path poem_dataset.pt --processes_num 16 \
55
  --seq_length 128 --target lm
56
  ```
 
58
  ```
59
  python3 pretrain.py --dataset_path poem_dataset.pt \
60
  --vocab_path models/google_zh_vocab.txt \
61
+ --output_model_path models/poem_gpt2_base_model.bin \
62
  --config_path models/bert_base_config.json \
63
  --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
64
  --total_steps 200000 --save_checkpoint_steps 50000 --report_steps 1000 \