Update README.md
Browse files
README.md
CHANGED
|
@@ -33,9 +33,9 @@ library_name: transformers
|
|
| 33 |
from fairseq.models.transformer import TransformerModel
|
| 34 |
```
|
| 35 |
* Load the model
|
| 36 |
-
```
|
| 37 |
-
model = TransformerModel.from_pretrained('path/to/model')
|
| 38 |
-
```
|
| 39 |
* Set the model to evaluation mode
|
| 40 |
```
|
| 41 |
model.eval()
|
|
@@ -50,10 +50,11 @@ model = TransformerModel.from_pretrained('path/to/model')
|
|
| 50 |
## Citation
|
| 51 |
|
| 52 |
**If you use our model, kindly cite our [paper](https://hal.science/hal-03430610/document)**:
|
| 53 |
-
|
| 54 |
-
@inproceedings{xu2021lisn,
|
| 55 |
title={LISN@ WMT 2021},
|
| 56 |
author={Xu, Jitao and Rauf, Sadaf Abdul and Pham, Minh Quang and Yvon, Fran{\c{c}}ois},
|
| 57 |
booktitle={6th Conference on Statistical Machine Translation},
|
| 58 |
year={2021}
|
| 59 |
-
}
|
|
|
|
|
|
| 33 |
from fairseq.models.transformer import TransformerModel
|
| 34 |
```
|
| 35 |
* Load the model
|
| 36 |
+
```
|
| 37 |
+
model = TransformerModel.from_pretrained('path/to/model')
|
| 38 |
+
```
|
| 39 |
* Set the model to evaluation mode
|
| 40 |
```
|
| 41 |
model.eval()
|
|
|
|
| 50 |
## Citation
|
| 51 |
|
| 52 |
**If you use our model, kindly cite our [paper](https://hal.science/hal-03430610/document)**:
|
| 53 |
+
```
|
| 54 |
+
@inproceedings{xu2021lisn,
|
| 55 |
title={LISN@ WMT 2021},
|
| 56 |
author={Xu, Jitao and Rauf, Sadaf Abdul and Pham, Minh Quang and Yvon, Fran{\c{c}}ois},
|
| 57 |
booktitle={6th Conference on Statistical Machine Translation},
|
| 58 |
year={2021}
|
| 59 |
+
}
|
| 60 |
+
```
|