motheecreator
commited on
Commit
•
73fc76f
1
Parent(s):
68bfe75
Update README.md
Browse files
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.
|
22 |
- Rouge2 Precision: None
|
23 |
- Rouge2 Recall: None
|
24 |
-
- Rouge2 Fmeasure: 0.
|
25 |
-
- Bleu: 9.
|
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.
|
60 |
-
| 2.
|
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
|