Model save
Browse files
README.md
CHANGED
@@ -1,8 +1,10 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
3 |
tags:
|
4 |
- generated_from_trainer
|
5 |
-
|
|
|
6 |
model-index:
|
7 |
- name: Vit-GPT2-COCO2017Flickr-85k-09
|
8 |
results: []
|
@@ -13,19 +15,14 @@ should probably proofread and complete it, then remove this comment. -->
|
|
13 |
|
14 |
# Vit-GPT2-COCO2017Flickr-85k-09
|
15 |
|
16 |
-
This model is a fine-tuned version of [NourFakih/Vit-GPT2-COCO2017Flickr-
|
17 |
It achieves the following results on the evaluation set:
|
18 |
-
-
|
19 |
-
-
|
20 |
-
-
|
21 |
-
-
|
22 |
-
-
|
23 |
-
-
|
24 |
-
- eval_runtime: 463.8252
|
25 |
-
- eval_samples_per_second: 8.624
|
26 |
-
- eval_steps_per_second: 2.156
|
27 |
-
- epoch: 0.0933
|
28 |
-
- step: 500
|
29 |
|
30 |
## Model description
|
31 |
|
@@ -54,6 +51,44 @@ The following hyperparameters were used during training:
|
|
54 |
- lr_scheduler_type: linear
|
55 |
- num_epochs: 3.0
|
56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
### Framework versions
|
58 |
|
59 |
- Transformers 4.41.2
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
base_model: NourFakih/Vit-GPT2-COCO2017Flickr-85k-09
|
4 |
tags:
|
5 |
- generated_from_trainer
|
6 |
+
metrics:
|
7 |
+
- rouge
|
8 |
model-index:
|
9 |
- name: Vit-GPT2-COCO2017Flickr-85k-09
|
10 |
results: []
|
|
|
15 |
|
16 |
# Vit-GPT2-COCO2017Flickr-85k-09
|
17 |
|
18 |
+
This model is a fine-tuned version of [NourFakih/Vit-GPT2-COCO2017Flickr-85k-09](https://huggingface.co/NourFakih/Vit-GPT2-COCO2017Flickr-85k-09) on an unknown dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.6343
|
21 |
+
- Rouge1: 38.8156
|
22 |
+
- Rouge2: 13.6737
|
23 |
+
- Rougel: 34.9479
|
24 |
+
- Rougelsum: 34.9604
|
25 |
+
- Gen Len: 12.1285
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
## Model description
|
28 |
|
|
|
51 |
- lr_scheduler_type: linear
|
52 |
- num_epochs: 3.0
|
53 |
|
54 |
+
### Training results
|
55 |
+
|
56 |
+
| Training Loss | Epoch | Step | Gen Len | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
57 |
+
|:-------------:|:------:|:-----:|:-------:|:---------------:|:-------:|:-------:|:-------:|:---------:|
|
58 |
+
| 0.2429 | 0.0933 | 500 | 11.738 | 0.5351 | 39.4446 | 14.1599 | 35.6123 | 35.5846 |
|
59 |
+
| 0.2537 | 0.1866 | 1000 | 12.3488 | 0.5301 | 39.5332 | 14.4745 | 35.644 | 35.6159 |
|
60 |
+
| 0.2564 | 0.2799 | 1500 | 12.2455 | 0.5198 | 39.8297 | 14.555 | 35.8598 | 35.8344 |
|
61 |
+
| 0.2585 | 0.3732 | 2000 | 11.8575 | 0.5207 | 39.4558 | 14.0496 | 35.5597 | 35.526 |
|
62 |
+
| 0.2579 | 0.4665 | 2500 | 11.9733 | 0.5188 | 39.1359 | 14.125 | 35.4068 | 35.3709 |
|
63 |
+
| 0.2588 | 0.5599 | 3000 | 12.278 | 0.5196 | 39.0831 | 14.0658 | 35.4608 | 35.4283 |
|
64 |
+
| 0.2618 | 0.6532 | 3500 | 11.9942 | 0.5194 | 39.751 | 14.443 | 36.076 | 36.0475 |
|
65 |
+
| 0.2579 | 0.7465 | 4000 | 12.0512 | 0.5102 | 39.7601 | 14.5095 | 36.0252 | 35.9857 |
|
66 |
+
| 0.2569 | 0.8398 | 4500 | 11.6483 | 0.5199 | 39.398 | 13.8871 | 35.7218 | 35.6911 |
|
67 |
+
| 0.253 | 0.9331 | 5000 | 12.0198 | 0.5200 | 39.8951 | 14.4146 | 35.883 | 35.8507 |
|
68 |
+
| 0.2361 | 1.0264 | 5500 | 12.183 | 0.5605 | 39.3352 | 14.2234 | 35.3107 | 35.2772 |
|
69 |
+
| 0.2 | 1.1197 | 6000 | 11.8598 | 0.5702 | 39.2184 | 14.0096 | 35.5475 | 35.5042 |
|
70 |
+
| 0.2034 | 1.2130 | 6500 | 11.878 | 0.5543 | 39.7118 | 14.2757 | 35.7613 | 35.7316 |
|
71 |
+
| 0.1968 | 1.3063 | 7000 | 12.1725 | 0.5584 | 39.1847 | 13.9003 | 35.3962 | 35.3713 |
|
72 |
+
| 0.1986 | 1.3996 | 7500 | 11.8395 | 0.5572 | 39.4428 | 14.2672 | 35.7359 | 35.7093 |
|
73 |
+
| 0.1988 | 1.4930 | 8000 | 11.9932 | 0.5552 | 39.2719 | 14.0411 | 35.482 | 35.4833 |
|
74 |
+
| 0.1971 | 1.5864 | 8500 | 12.1003 | 0.5572 | 39.2681 | 14.1036 | 35.4466 | 35.4245 |
|
75 |
+
| 0.1978 | 1.6797 | 9000 | 12.1152 | 0.5667 | 39.2673 | 14.0918 | 35.4179 | 35.4169 |
|
76 |
+
| 0.1937 | 1.7730 | 9500 | 12.2208 | 0.5781 | 39.4115 | 14.1115 | 35.6952 | 35.6834 |
|
77 |
+
| 0.1897 | 1.8663 | 10000 | 11.8818 | 0.5754 | 39.2059 | 14.076 | 35.3392 | 35.3332 |
|
78 |
+
| 0.1898 | 1.9596 | 10500 | 11.8818 | 0.5720 | 39.4033 | 14.1447 | 35.598 | 35.5976 |
|
79 |
+
| 0.1685 | 2.0529 | 11000 | 12.0585 | 0.6186 | 38.4626 | 13.4695 | 34.7378 | 34.7294 |
|
80 |
+
| 0.1454 | 2.1462 | 11500 | 11.9448 | 0.6147 | 38.5335 | 13.5152 | 34.7075 | 34.7033 |
|
81 |
+
| 0.1434 | 2.2395 | 12000 | 12.1855 | 0.6229 | 39.0044 | 13.9276 | 35.2226 | 35.2116 |
|
82 |
+
| 0.1479 | 2.3328 | 12500 | 12.0273 | 0.6262 | 38.6281 | 13.5737 | 34.8247 | 34.8245 |
|
83 |
+
| 0.1452 | 2.4261 | 13000 | 12.0222 | 0.6243 | 38.9136 | 13.6727 | 35.0597 | 35.0643 |
|
84 |
+
| 0.1464 | 2.5195 | 13500 | 12.006 | 0.6309 | 38.9915 | 13.5041 | 34.9971 | 34.9991 |
|
85 |
+
| 0.1431 | 2.6128 | 14000 | 12.0602 | 0.6318 | 38.7595 | 13.5585 | 34.8308 | 34.834 |
|
86 |
+
| 0.1431 | 2.7061 | 14500 | 12.229 | 0.6277 | 38.8899 | 13.7343 | 34.9536 | 34.9513 |
|
87 |
+
| 0.1445 | 2.7995 | 15000 | 12.0343 | 0.6357 | 38.7681 | 13.5849 | 34.9764 | 34.9564 |
|
88 |
+
| 0.1379 | 2.8928 | 15500 | 0.6340 | 38.9196 | 13.6285 | 34.9761 | 34.9855 | 12.1242 |
|
89 |
+
| 0.1411 | 2.9861 | 16000 | 0.6343 | 38.8156 | 13.6737 | 34.9479 | 34.9604 | 12.1285 |
|
90 |
+
|
91 |
+
|
92 |
### Framework versions
|
93 |
|
94 |
- Transformers 4.41.2
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 956835520
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e5da89b433959f2c23b679f74129de4c8da1e30d5b601691eb0b612d2149ec87
|
3 |
size 956835520
|
runs/Jun21_00-00-49_21cfc91cdd14/events.out.tfevents.1718928052.21cfc91cdd14.34.0
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:97c7cd7a74f7623520bfe2c8199fa50edb049ecc5672242e95b215722a1dd79f
|
3 |
+
size 11106
|