Compare commits
10 Commits
73fe2515ab
...
39fdcea592
Author | SHA1 | Date |
---|---|---|
|
39fdcea592 | |
|
bc9a8cd897 | |
|
a06887d8d7 | |
|
addbaed520 | |
|
1f663e796e | |
|
ea3b5bfee3 | |
|
92702f0b92 | |
|
8dd4da7142 | |
|
ee167a9ab5 | |
|
103015292d |
82
README.md
82
README.md
|
@ -1,22 +1,65 @@
|
|||
---
|
||||
tags:
|
||||
- translation
|
||||
license: cc-by-4.0
|
||||
---
|
||||
|
||||
### opus-mt-ru-en
|
||||
|
||||
* source languages: ru
|
||||
* target languages: en
|
||||
* OPUS readme: [ru-en](https://github.com/Helsinki-NLP/OPUS-MT-train/blob/master/models/ru-en/README.md)
|
||||
## Table of Contents
|
||||
- [Model Details](#model-details)
|
||||
- [Uses](#uses)
|
||||
- [Risks, Limitations and Biases](#risks-limitations-and-biases)
|
||||
- [Training](#training)
|
||||
- [Evaluation](#evaluation)
|
||||
- [Citation Information](#citation-information)
|
||||
- [How to Get Started With the Model](#how-to-get-started-with-the-model)
|
||||
|
||||
## Model Details
|
||||
**Model Description:**
|
||||
- **Developed by:** Language Technology Research Group at the University of Helsinki
|
||||
- **Model Type:** Transformer-align
|
||||
- **Language(s):**
|
||||
- Source Language: Russian
|
||||
- Target Language: English
|
||||
- **License:** CC-BY-4.0
|
||||
- **Resources for more information:**
|
||||
- [GitHub Repo](https://github.com/Helsinki-NLP/OPUS-MT-train)
|
||||
|
||||
|
||||
|
||||
## Uses
|
||||
|
||||
#### Direct Use
|
||||
|
||||
This model can be used for translation and text-to-text generation.
|
||||
|
||||
|
||||
## Risks, Limitations and Biases
|
||||
|
||||
**CONTENT WARNING: Readers should be aware this section contains content that is disturbing, offensive, and can propagate historical and current stereotypes.**
|
||||
|
||||
Significant research has explored bias and fairness issues with language models (see, e.g., [Sheng et al. (2021)](https://aclanthology.org/2021.acl-long.330.pdf) and [Bender et al. (2021)](https://dl.acm.org/doi/pdf/10.1145/3442188.3445922)).
|
||||
|
||||
Further details about the dataset for this model can be found in the OPUS readme: [ru-en](https://github.com/Helsinki-NLP/OPUS-MT-train/blob/master/models/ru-en/README.md)
|
||||
|
||||
## Training
|
||||
#### Training Data
|
||||
##### Preprocessing
|
||||
* Pre-processing: Normalization + SentencePiece
|
||||
* Dataset: [opus](https://github.com/Helsinki-NLP/Opus-MT)
|
||||
* Download original weights: [opus-2020-02-26.zip](https://object.pouta.csc.fi/OPUS-MT-models/ru-en/opus-2020-02-26.zip)
|
||||
|
||||
* Test set translations: [opus-2020-02-26.test.txt](https://object.pouta.csc.fi/OPUS-MT-models/ru-en/opus-2020-02-26.test.txt)
|
||||
|
||||
|
||||
## Evaluation
|
||||
|
||||
#### Results
|
||||
|
||||
* dataset: opus
|
||||
* model: transformer-align
|
||||
* pre-processing: normalization + SentencePiece
|
||||
* download original weights: [opus-2020-02-26.zip](https://object.pouta.csc.fi/OPUS-MT-models/ru-en/opus-2020-02-26.zip)
|
||||
* test set translations: [opus-2020-02-26.test.txt](https://object.pouta.csc.fi/OPUS-MT-models/ru-en/opus-2020-02-26.test.txt)
|
||||
* test set scores: [opus-2020-02-26.eval.txt](https://object.pouta.csc.fi/OPUS-MT-models/ru-en/opus-2020-02-26.eval.txt)
|
||||
|
||||
## Benchmarks
|
||||
#### Benchmarks
|
||||
|
||||
| testset | BLEU | chr-F |
|
||||
|-----------------------|-------|-------|
|
||||
|
@ -30,3 +73,24 @@ tags:
|
|||
| newstest2019-ruen.ru.en | 31.4 | 0.576 |
|
||||
| Tatoeba.ru.en | 61.1 | 0.736 |
|
||||
|
||||
## Citation Information
|
||||
|
||||
```bibtex
|
||||
@InProceedings{TiedemannThottingal:EAMT2020,
|
||||
author = {J{\"o}rg Tiedemann and Santhosh Thottingal},
|
||||
title = {{OPUS-MT} — {B}uilding open translation services for the {W}orld},
|
||||
booktitle = {Proceedings of the 22nd Annual Conferenec of the European Association for Machine Translation (EAMT)},
|
||||
year = {2020},
|
||||
address = {Lisbon, Portugal}
|
||||
}
|
||||
```
|
||||
|
||||
## How to Get Started With the Model
|
||||
|
||||
```python
|
||||
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
||||
|
||||
tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
|
||||
|
||||
model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
|
||||
```
|
||||
|
|
|
@ -1,4 +1,5 @@
|
|||
{
|
||||
"_name_or_path": "/tmp/Helsinki-NLP/opus-mt-ru-en",
|
||||
"_num_labels": 3,
|
||||
"activation_dropout": 0.0,
|
||||
"activation_function": "swish",
|
||||
|
@ -15,18 +16,21 @@
|
|||
],
|
||||
"bos_token_id": 0,
|
||||
"classif_dropout": 0.0,
|
||||
"classifier_dropout": 0.0,
|
||||
"d_model": 512,
|
||||
"decoder_attention_heads": 8,
|
||||
"decoder_ffn_dim": 2048,
|
||||
"decoder_layerdrop": 0.0,
|
||||
"decoder_layers": 6,
|
||||
"decoder_start_token_id": 62517,
|
||||
"decoder_vocab_size": 62518,
|
||||
"dropout": 0.1,
|
||||
"encoder_attention_heads": 8,
|
||||
"encoder_ffn_dim": 2048,
|
||||
"encoder_layerdrop": 0.0,
|
||||
"encoder_layers": 6,
|
||||
"eos_token_id": 0,
|
||||
"forced_eos_token_id": 0,
|
||||
"id2label": {
|
||||
"0": "LABEL_0",
|
||||
"1": "LABEL_1",
|
||||
|
@ -48,6 +52,9 @@
|
|||
"num_hidden_layers": 6,
|
||||
"pad_token_id": 62517,
|
||||
"scale_embedding": true,
|
||||
"share_encoder_decoder_embeddings": true,
|
||||
"static_position_embeddings": true,
|
||||
"transformers_version": "4.22.0.dev0",
|
||||
"use_cache": true,
|
||||
"vocab_size": 62518
|
||||
}
|
||||
|
|
|
@ -0,0 +1,16 @@
|
|||
{
|
||||
"_from_model_config": true,
|
||||
"bad_words_ids": [
|
||||
[
|
||||
62517
|
||||
]
|
||||
],
|
||||
"bos_token_id": 0,
|
||||
"decoder_start_token_id": 62517,
|
||||
"eos_token_id": 0,
|
||||
"forced_eos_token_id": 0,
|
||||
"max_length": 512,
|
||||
"num_beams": 6,
|
||||
"pad_token_id": 62517,
|
||||
"transformers_version": "4.27.0.dev0"
|
||||
}
|
Binary file not shown.
Loading…
Reference in New Issue