End of training

README.md

@@ -5,7 +5,6 @@ tags:
 model-index:
 - name: Auto-pharmacy-V3
   results: []
-pipeline_tag: image-to-text
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -40,7 +39,7 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 1
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
 ### Framework versions
@@ -48,4 +47,4 @@ The following hyperparameters were used during training:
 - Transformers 4.39.3
 - Pytorch 2.1.2
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2

config.json

@@ -81,9 +81,10 @@
     "use_bfloat16": false,
     "use_cache": false,
     "use_learned_position_embeddings": true,
-    "vocab_size": 64044
+    "vocab_size": 50265
   },
   "decoder_start_token_id": 0,
+  "early_stopping": true,
   "encoder": {
     "_name_or_path": "",
     "add_cross_attention": false,
@@ -163,9 +164,14 @@
   },
   "eos_token_id": 2,
   "is_encoder_decoder": true,
+  "length_penalty": 2.0,
+  "max_length": 11,
   "model_type": "vision-encoder-decoder",
+  "no_repeat_ngram_size": 3,
+  "num_beams": 8,
   "pad_token_id": 1,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.3"
+  "transformers_version": "4.39.3",
+  "vocab_size": 50265
 }

generation_config.json

@@ -1,8 +1,13 @@
 {
   "_from_model_config": true,
   "bos_token_id": 0,
-  "decoder_start_token_id": 2,
+  "decoder_start_token_id": 0,
+  "early_stopping": true,
   "eos_token_id": 2,
+  "length_penalty": 2.0,
+  "max_length": 11,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 8,
   "pad_token_id": 1,
   "transformers_version": "4.39.3",
   "use_cache": false

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cc2e35eac96d01cc46277959afa1c09067399ef495b47a745d723fcf2ddc684e
-size 246430696
+oid sha256:92309973080c1e1a42d856ba651dec7767cdba4f5507ebddae04e2ec4cebe821
+size 218412460

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8be483e22de4bd4aa70b490f0c953632049122e2ca8e063f9b6fe018d8845301
+oid sha256:1240abebcd17ba83fc49d694d73c06e15c35f9775eab27fe87e40a22fbef7861
 size 4920