joheras commited on
Commit
ee89fab
verified
1 Parent(s): ca7a5aa

Training complete

Browse files
Files changed (2) hide show
  1. README.md +11 -10
  2. generation_config.json +1 -1
README.md CHANGED
@@ -1,4 +1,5 @@
1
  ---
 
2
  license: mit
3
  base_model: facebook/mbart-large-50
4
  tags:
@@ -18,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0220
22
- - Bleu: 98.2132
23
- - Gen Len: 18.5417
24
 
25
  ## Model description
26
 
@@ -43,7 +44,7 @@ The following hyperparameters were used during training:
43
  - train_batch_size: 8
44
  - eval_batch_size: 8
45
  - seed: 42
46
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - num_epochs: 2
49
 
@@ -51,13 +52,13 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
54
- | No log | 1.0 | 440 | 0.0490 | 96.2659 | 19.0104 |
55
- | 0.2462 | 2.0 | 880 | 0.0220 | 98.2132 | 18.5417 |
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.38.1
61
- - Pytorch 2.1.0+cu121
62
- - Datasets 2.17.1
63
- - Tokenizers 0.15.2
 
1
  ---
2
+ library_name: transformers
3
  license: mit
4
  base_model: facebook/mbart-large-50
5
  tags:
 
19
 
20
  This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.0134
23
+ - Bleu: 98.9481
24
+ - Gen Len: 18.6354
25
 
26
  ## Model description
27
 
 
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 42
47
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
48
  - lr_scheduler_type: linear
49
  - num_epochs: 2
50
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
55
+ | No log | 1.0 | 440 | 0.0162 | 97.6428 | 18.5417 |
56
+ | 0.2252 | 2.0 | 880 | 0.0134 | 98.9481 | 18.6354 |
57
 
58
 
59
  ### Framework versions
60
 
61
+ - Transformers 4.47.1
62
+ - Pytorch 2.5.1+cu121
63
+ - Datasets 3.2.0
64
+ - Tokenizers 0.21.0
generation_config.json CHANGED
@@ -7,5 +7,5 @@
7
  "max_length": 200,
8
  "num_beams": 5,
9
  "pad_token_id": 1,
10
- "transformers_version": "4.38.1"
11
  }
 
7
  "max_length": 200,
8
  "num_beams": 5,
9
  "pad_token_id": 1,
10
+ "transformers_version": "4.47.1"
11
  }