End of training

README.md

@@ -1,9 +1,10 @@
 ---
 library_name: transformers
+base_model: motheecreator/ViT-GPT2-Image_Captioning_model
 tags:
 - generated_from_trainer
-- image-to-text
-- image-captioning
+metrics:
+- bleu
 model-index:
 - name: ViT-GPT2
   results: []
@@ -14,10 +15,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ViT-GPT2
 
-This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+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.
 It achieves the following results on the evaluation set:
-- Loss: 2.4134
-- Rouge2 Fmeasure: 0.1166
+- Loss: 2.1879
+- Rouge2 Precision: None
+- Rouge2 Recall: None
+- Rouge2 Fmeasure: 0.1506
+- Bleu: 9.3133
 
 ## Model description
 
@@ -44,16 +48,15 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 256
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 3.0
+- num_epochs: 2
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch  | Step | Validation Loss | Rouge2 Fmeasure |
-|:-------------:|:------:|:----:|:---------------:|:---------------:|
-| No log        | 0.9987 | 496  | 2.4901          | 0.1077          |
-| 2.5089        | 1.9995 | 993  | 2.4292          | 0.1141          |
-| 2.4103        | 2.9962 | 1488 | 2.4134          | 0.1166          |
+| Training Loss | Epoch  | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure | Bleu   |
+|:-------------:|:------:|:----:|:---------------:|:----------------:|:-------------:|:---------------:|:------:|
+| 2.2959        | 0.9993 | 1171 | 2.2239          | None             | None          | 0.1474          | 8.9628 |
+| 2.1491        | 1.9985 | 2342 | 2.1879          | None             | None          | 0.1506          | 9.3133 |
 
 
 ### Framework versions
@@ -61,4 +64,4 @@ The following hyperparameters were used during training:
 - Transformers 4.44.2
 - Pytorch 2.4.0
 - Datasets 3.0.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "motheecreator/ViT-GPT2-Image_Captioning_model",
   "architectures": [
     "VisionEncoderDecoderModel"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be7a4e893197f6ccb16f8ffc4de64cc65e716748e55403a160f8beb2ec83e36b
+oid sha256:df6d7cacba8005a9958206742d9a5927c35137802cf52daebc4f41a431f54517
 size 956835520

runs/Sep30_11-07-05_bc37da79282d/events.out.tfevents.1727694439.bc37da79282d.30.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c5c9085c16718ab4b6e7cf3ce9fe536cef2737fe3ce7c4bb9750ce0c67a555b
+size 11281

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0bee922a139f7bb463929ce1596c39f316085f09c6abce09f9339d0ec028e22d
+oid sha256:fbaa3012913b518ee4ffc38680bb18e9c7c709d6b2a11823544cebeb6878c105
 size 5304