Compare commits

...

10 Commits

Author SHA1 Message Date
Joao Gante 39fdcea592 Adding generation config file(s) 2023-01-24 15:41:29 +00:00
Joao Gante bc9a8cd897 Add TF weights 2022-09-05 12:00:27 +00:00
Jörg Tiedemann a06887d8d7 Update README.md 2022-07-14 08:56:05 +00:00
tiedeman addbaed520 Model Card (#1)
- Model Card (ed0317f25efd4f02ddd14874740600b267cb3a76)


Co-authored-by: Ezi Ozoani <Ezi@users.noreply.huggingface.co>
2022-07-14 08:55:36 +00:00
Julien Chaumond 1f663e796e metadata: add license 2021-09-10 16:02:18 +02:00
system ea3b5bfee3 Update README.md 2020-08-21 14:42:49 +00:00
system 92702f0b92 Update README.md 2020-08-19 00:44:20 +00:00
system 8dd4da7142 Update config.json 2020-08-19 00:43:55 +00:00
system ee167a9ab5 Update tokenizer_config.json 2020-08-19 00:43:55 +00:00
system 103015292d Update vocab.json 2020-08-19 00:43:55 +00:00
4 changed files with 99 additions and 9 deletions

View File

@ -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")
```

View File

@ -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
}

16
generation_config.json Normal file
View File

@ -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"
}

BIN
tf_model.h5 (Stored with Git LFS) Normal file

Binary file not shown.