Hugging Face's logo

ChayanM
/
ViT-Bert_Mimic

Image-Text-to-Text
Transformers
TensorBoard
Safetensors
vision-encoder-decoder
Generated from Trainer
Model card Files
xet
 Metrics Community
ChayanM commited on
Commit
e6503e1
·
verified ·
1 Parent(s): ded05be

Model save

Browse files
Files changed (2) hide show
  1. README.md +9 -9
  2. generation_config.json +1 -0
README.md CHANGED
@@ -15,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.2334
19
- - Rouge1: 27.4942
20
- - Rouge2: 21.9051
21
- - Rougel: 26.9521
22
- - Rougelsum: 27.0238
23
- - Gen Len: 16.0
24
 
25
  ## Model description
26
 
@@ -51,9 +51,9 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
54
- | No log | 1.0 | 125 | 0.3533 | 0.0 | 0.0 | 0.0 | 0.0 | 20.0 |
55
- | No log | 2.0 | 250 | 0.3009 | 7.9737 | 1.5 | 7.7593 | 7.9222 | 16.38 |
56
- | No log | 3.0 | 375 | 0.2334 | 27.4942 | 21.9051 | 26.9521 | 27.0238 | 16.0 |
57
 
58
 
59
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.1679
19
+ - Rouge1: 33.2921
20
+ - Rouge2: 26.4362
21
+ - Rougel: 33.3429
22
+ - Rougelsum: 33.528
23
+ - Gen Len: 15.06
24
 
25
  ## Model description
26
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
54
+ | No log | 1.0 | 125 | 0.2445 | 0.0 | 0.0 | 0.0 | 0.0 | 200.0 |
55
+ | No log | 2.0 | 250 | 0.1816 | 33.2921 | 26.4362 | 33.3429 | 33.528 | 12.04 |
56
+ | No log | 3.0 | 375 | 0.1679 | 33.2921 | 26.4362 | 33.3429 | 33.528 | 15.06 |
57
 
58
 
59
  ### Framework versions
generation_config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "decoder_start_token_id": 101,
3
  "eos_token_id": 102,
 
4
  "pad_token_id": 0,
5
  "transformers_version": "4.37.1"
6
  }
 
1
  {
2
  "decoder_start_token_id": 101,
3
  "eos_token_id": 102,
4
+ "max_length": 200,
5
  "pad_token_id": 0,
6
  "transformers_version": "4.37.1"
7
  }