Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ python3 preprocess.py --corpus_path corpora/lyric.txt \
|
|
43 |
|
44 |
```
|
45 |
python3 pretrain.py --dataset_path lyric_dataset.pt \
|
46 |
-
|
47 |
--vocab_path models/google_zh_vocab.txt \
|
48 |
--output_model_path models/lyric_gpt2_seq512_model.bin \
|
49 |
--config_path models/bert_base_config.json --learning_rate 5e-5 \
|
|
|
43 |
|
44 |
```
|
45 |
python3 pretrain.py --dataset_path lyric_dataset.pt \
|
46 |
+
--pretrained_model_path gpt2-base-chinese-cluecorpussmall/pytorch_model.bin\
|
47 |
--vocab_path models/google_zh_vocab.txt \
|
48 |
--output_model_path models/lyric_gpt2_seq512_model.bin \
|
49 |
--config_path models/bert_base_config.json --learning_rate 5e-5 \
|