Update README.md
Browse files
README.md
CHANGED
@@ -63,7 +63,7 @@ python3 pretrain.py --dataset_path cluecorpussmall_pegasus_seq512_dataset.pt \
|
|
63 |
Finally, we convert the pre-trained model into Huggingface's format:
|
64 |
|
65 |
```
|
66 |
-
python3 scripts/convert_pegasus_from_uer_to_huggingface.py --input_model_path cluecorpussmall_pegasus_base_seq512_model.bin-
|
67 |
--output_model_path pytorch_model.bin \
|
68 |
--layers_num 12
|
69 |
```
|
|
|
63 |
Finally, we convert the pre-trained model into Huggingface's format:
|
64 |
|
65 |
```
|
66 |
+
python3 scripts/convert_pegasus_from_uer_to_huggingface.py --input_model_path models/cluecorpussmall_pegasus_base_seq512_model.bin-1000000 \
|
67 |
--output_model_path pytorch_model.bin \
|
68 |
--layers_num 12
|
69 |
```
|