Update README.md
Browse files
README.md
CHANGED
@@ -70,7 +70,7 @@ python3 pretrain.py --dataset_path couplet_dataset.pt \
|
|
70 |
|
71 |
Finally, we convert the pre-trained model into Huggingface's format:
|
72 |
```
|
73 |
-
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path couplet_gpt2_model.bin-25000 \
|
74 |
--output_model_path pytorch_model.bin \
|
75 |
--layers_num 12
|
76 |
```
|
|
|
70 |
|
71 |
Finally, we convert the pre-trained model into Huggingface's format:
|
72 |
```
|
73 |
+
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path models/couplet_gpt2_model.bin-25000 \
|
74 |
--output_model_path pytorch_model.bin \
|
75 |
--layers_num 12
|
76 |
```
|