yasmineelabbar commited on
Commit
34cbd7d
1 Parent(s): e5e0736

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -3
README.md CHANGED
@@ -5,9 +5,24 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - opus_books
 
 
8
  model-index:
9
  - name: my_awesome_opus_books_model
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +31,10 @@ should probably proofread and complete it, then remove this comment. -->
16
  # my_awesome_opus_books_model
17
 
18
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the opus_books dataset.
 
 
 
 
19
 
20
  ## Model description
21
 
@@ -35,13 +54,21 @@ More information needed
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
- - train_batch_size: 16
39
- - eval_batch_size: 16
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
  - num_epochs: 2
44
 
 
 
 
 
 
 
 
 
45
  ### Framework versions
46
 
47
  - Transformers 4.31.0
 
5
  - generated_from_trainer
6
  datasets:
7
  - opus_books
8
+ metrics:
9
+ - bleu
10
  model-index:
11
  - name: my_awesome_opus_books_model
12
+ results:
13
+ - task:
14
+ name: Sequence-to-sequence Language Modeling
15
+ type: text2text-generation
16
+ dataset:
17
+ name: opus_books
18
+ type: opus_books
19
+ config: en-fr
20
+ split: train
21
+ args: en-fr
22
+ metrics:
23
+ - name: Bleu
24
+ type: bleu
25
+ value: 5.4949
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
  # my_awesome_opus_books_model
32
 
33
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the opus_books dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 1.6367
36
+ - Bleu: 5.4949
37
+ - Gen Len: 17.6014
38
 
39
  ## Model description
40
 
 
54
 
55
  The following hyperparameters were used during training:
56
  - learning_rate: 2e-05
57
+ - train_batch_size: 32
58
+ - eval_batch_size: 32
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
  - num_epochs: 2
63
 
64
+ ### Training results
65
+
66
+ | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
67
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|
68
+ | 1.8908 | 1.0 | 3178 | 1.6597 | 5.3117 | 17.6181 |
69
+ | 1.8545 | 2.0 | 6356 | 1.6367 | 5.4949 | 17.6014 |
70
+
71
+
72
  ### Framework versions
73
 
74
  - Transformers 4.31.0