Update README.md

README.md

@@ -26,12 +26,12 @@ The training was conducted with the following hyperparameters:
 
 - base model: [google/mt5-small](https://huggingface.co/google/mt5-small)
 - source_prefix: `"summarize: "`
-- batch size: xxx
+- batch size: 3 (6)
 - max_source_length: 800
 - max_target_length: 96
 - warmup_ratio: 0.3
-- number of train epochs: xxx
-- gradient accumulation steps: xxx
+- number of train epochs: 10
+- gradient accumulation steps: 2
 
 ## Datasets and Preprocessing
 
@@ -49,7 +49,7 @@ This model is trained on the following dataset:
 
 | Model | rouge1 | rouge2 | rougeL | rougeLsum
 |-------|--------|--------|--------|----------
-| deutsche-telekom/mt5-small-sum-de-mit-v1 (this) | xxx | xxx | xxx | xxx
+| deutsche-telekom/mt5-small-sum-de-mit-v1 (this) | 16.8023 | 3.5531 | 12.6884 | 14.7624
 | [ml6team/mt5-small-german-finetune-mlsum](https://huggingface.co/ml6team/mt5-small-german-finetune-mlsum) | 18.3607 | 5.3604 | 14.5456 | 16.1946
 | **[deutsche-telekom/mt5-small-sum-de-en-01](https://huggingface.co/deutsche-telekom/mt5-small-sum-de-en-v1)** | **21.7336** | **7.2614** | **17.1323** | **19.3977**