Mohammed Abdeldayem
commited on
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
|