motheecreator commited on
Commit
73fc76f
1 Parent(s): 68bfe75

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [motheecreator/ViT-GPT2-Image_Captioning_model](https://huggingface.co/motheecreator/ViT-GPT2-Image_Captioning_model) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 2.1879
22
  - Rouge2 Precision: None
23
  - Rouge2 Recall: None
24
- - Rouge2 Fmeasure: 0.1506
25
- - Bleu: 9.3133
26
 
27
  ## Model description
28
 
@@ -56,8 +56,8 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure | Bleu |
58
  |:-------------:|:------:|:----:|:---------------:|:----------------:|:-------------:|:---------------:|:------:|
59
- | 2.2959 | 0.9993 | 1171 | 2.2239 | None | None | 0.1474 | 8.9628 |
60
- | 2.1491 | 1.9985 | 2342 | 2.1879 | None | None | 0.1506 | 9.3133 |
61
 
62
 
63
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [motheecreator/ViT-GPT2-Image_Captioning_model](https://huggingface.co/motheecreator/ViT-GPT2-Image_Captioning_model) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 2.125337
22
  - Rouge2 Precision: None
23
  - Rouge2 Recall: None
24
+ - Rouge2 Fmeasure: 0.155
25
+ - Bleu: 9.7054
26
 
27
  ## Model description
28
 
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure | Bleu |
58
  |:-------------:|:------:|:----:|:---------------:|:----------------:|:-------------:|:---------------:|:------:|
59
+ | 2.1537 | 0.9993 | 1171 | 2.13666 | None | None | 0.1531 | 9.4673 |
60
+ | 2.0434 | 1.9985 | 2342 | 2.125337 | None | None | 0.155 | 9.7054 |
61
 
62
 
63
  ### Framework versions