Update README.md
Browse files
README.md
CHANGED
@@ -91,7 +91,7 @@ python3 pretrain.py --dataset_path cluecorpussmall_lm_seq1024_dataset.pt \
|
|
91 |
Finally, we convert the pre-trained model into Huggingface's format:
|
92 |
|
93 |
```
|
94 |
-
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path cluecorpussmall_gpt2_distil_seq1024_model.bin-250000 \
|
95 |
--output_model_path pytorch_model.bin \
|
96 |
--layers_num 6
|
97 |
```
|
|
|
91 |
Finally, we convert the pre-trained model into Huggingface's format:
|
92 |
|
93 |
```
|
94 |
+
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path models/cluecorpussmall_gpt2_distil_seq1024_model.bin-250000 \
|
95 |
--output_model_path pytorch_model.bin \
|
96 |
--layers_num 6
|
97 |
```
|