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
4e11c2f
•
1 Parent(s): 46720b8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -58,8 +58,8 @@ 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 \
65
  --learning_rate 5e-4 --batch_size 64 \
 
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_model.bin \
62
+ --config_path models/gpt2/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 \
65
  --learning_rate 5e-4 --batch_size 64 \