Winmodel commited on
Commit
65f4d38
·
verified ·
1 Parent(s): 775d8bc

End of training

Browse files
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ metrics:
5
+ - rouge
6
+ model-index:
7
+ - name: image-captioning-output
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # image-captioning-output
15
+
16
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.2689
19
+ - Rouge1: 0.0
20
+ - Rouge2: 0.0
21
+ - Rougel: 0.0
22
+ - Rougelsum: 0.0
23
+ - Gen Len: 9.707
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 5e-05
43
+ - train_batch_size: 16
44
+ - eval_batch_size: 32
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - num_epochs: 3.0
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
53
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
54
+ | No log | 1.0 | 375 | 0.3010 | 0.0 | 0.0 | 0.0 | 0.0 | 10.073 |
55
+ | 0.391 | 2.0 | 750 | 0.2773 | 0.0 | 0.0 | 0.0 | 0.0 | 12.193 |
56
+ | 0.2852 | 3.0 | 1125 | 0.2689 | 0.0 | 0.0 | 0.0 | 0.0 | 9.707 |
57
+
58
+
59
+ ### Framework versions
60
+
61
+ - Transformers 4.38.2
62
+ - Pytorch 2.2.1+cu121
63
+ - Datasets 2.19.0
64
+ - Tokenizers 0.15.2
config.json ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "VisionEncoderDecoderModel"
4
+ ],
5
+ "decoder": {
6
+ "_name_or_path": "openthaigpt/openthaigpt-gpt2-instructgpt-poc-0.0.4",
7
+ "activation_function": "gelu_new",
8
+ "add_cross_attention": true,
9
+ "architectures": [
10
+ "GPT2LMHeadModel"
11
+ ],
12
+ "attn_pdrop": 0.0,
13
+ "bad_words_ids": null,
14
+ "begin_suppress_tokens": null,
15
+ "bos_token_id": 50256,
16
+ "chunk_size_feed_forward": 0,
17
+ "cross_attention_hidden_size": null,
18
+ "decoder_start_token_id": null,
19
+ "diversity_penalty": 0.0,
20
+ "do_sample": false,
21
+ "early_stopping": false,
22
+ "embd_pdrop": 0.0,
23
+ "encoder_no_repeat_ngram_size": 0,
24
+ "eos_token_id": 50256,
25
+ "exponential_decay_length_penalty": null,
26
+ "finetuning_task": null,
27
+ "forced_bos_token_id": null,
28
+ "forced_eos_token_id": null,
29
+ "gradient_checkpointing": false,
30
+ "id2label": {
31
+ "0": "LABEL_0",
32
+ "1": "LABEL_1"
33
+ },
34
+ "initializer_range": 0.02,
35
+ "is_decoder": true,
36
+ "is_encoder_decoder": false,
37
+ "label2id": {
38
+ "LABEL_0": 0,
39
+ "LABEL_1": 1
40
+ },
41
+ "layer_norm_epsilon": 1e-05,
42
+ "length_penalty": 1.0,
43
+ "max_length": 20,
44
+ "min_length": 0,
45
+ "model_type": "gpt2",
46
+ "n_ctx": 1024,
47
+ "n_embd": 768,
48
+ "n_head": 12,
49
+ "n_inner": null,
50
+ "n_layer": 12,
51
+ "n_positions": 1024,
52
+ "no_repeat_ngram_size": 0,
53
+ "num_beam_groups": 1,
54
+ "num_beams": 1,
55
+ "num_return_sequences": 1,
56
+ "output_attentions": false,
57
+ "output_hidden_states": false,
58
+ "output_scores": false,
59
+ "pad_token_id": null,
60
+ "prefix": null,
61
+ "problem_type": null,
62
+ "pruned_heads": {},
63
+ "remove_invalid_values": false,
64
+ "reorder_and_upcast_attn": false,
65
+ "repetition_penalty": 1.0,
66
+ "resid_pdrop": 0.0,
67
+ "return_dict": true,
68
+ "return_dict_in_generate": false,
69
+ "scale_attn_by_inverse_layer_idx": false,
70
+ "scale_attn_weights": true,
71
+ "sep_token_id": null,
72
+ "summary_activation": null,
73
+ "summary_first_dropout": 0.1,
74
+ "summary_proj_to_labels": true,
75
+ "summary_type": "cls_index",
76
+ "summary_use_proj": true,
77
+ "suppress_tokens": null,
78
+ "task_specific_params": {
79
+ "text-generation": {
80
+ "do_sample": true,
81
+ "max_length": 50
82
+ }
83
+ },
84
+ "temperature": 1.0,
85
+ "tf_legacy_loss": false,
86
+ "tie_encoder_decoder": false,
87
+ "tie_word_embeddings": true,
88
+ "tokenizer_class": null,
89
+ "top_k": 50,
90
+ "top_p": 1.0,
91
+ "torch_dtype": "float32",
92
+ "torchscript": false,
93
+ "typical_p": 1.0,
94
+ "use_bfloat16": false,
95
+ "use_cache": true,
96
+ "vocab_size": 50268
97
+ },
98
+ "decoder_start_token_id": 50266,
99
+ "encoder": {
100
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
101
+ "add_cross_attention": false,
102
+ "architectures": [
103
+ "ViTModel"
104
+ ],
105
+ "attention_probs_dropout_prob": 0.0,
106
+ "bad_words_ids": null,
107
+ "begin_suppress_tokens": null,
108
+ "bos_token_id": null,
109
+ "chunk_size_feed_forward": 0,
110
+ "cross_attention_hidden_size": null,
111
+ "decoder_start_token_id": null,
112
+ "diversity_penalty": 0.0,
113
+ "do_sample": false,
114
+ "early_stopping": false,
115
+ "encoder_no_repeat_ngram_size": 0,
116
+ "encoder_stride": 16,
117
+ "eos_token_id": null,
118
+ "exponential_decay_length_penalty": null,
119
+ "finetuning_task": null,
120
+ "forced_bos_token_id": null,
121
+ "forced_eos_token_id": null,
122
+ "hidden_act": "gelu",
123
+ "hidden_dropout_prob": 0.0,
124
+ "hidden_size": 768,
125
+ "id2label": {
126
+ "0": "LABEL_0",
127
+ "1": "LABEL_1"
128
+ },
129
+ "image_size": 224,
130
+ "initializer_range": 0.02,
131
+ "intermediate_size": 3072,
132
+ "is_decoder": false,
133
+ "is_encoder_decoder": false,
134
+ "label2id": {
135
+ "LABEL_0": 0,
136
+ "LABEL_1": 1
137
+ },
138
+ "layer_norm_eps": 1e-12,
139
+ "length_penalty": 1.0,
140
+ "max_length": 20,
141
+ "min_length": 0,
142
+ "model_type": "vit",
143
+ "no_repeat_ngram_size": 0,
144
+ "num_attention_heads": 12,
145
+ "num_beam_groups": 1,
146
+ "num_beams": 1,
147
+ "num_channels": 3,
148
+ "num_hidden_layers": 12,
149
+ "num_return_sequences": 1,
150
+ "output_attentions": false,
151
+ "output_hidden_states": false,
152
+ "output_scores": false,
153
+ "pad_token_id": null,
154
+ "patch_size": 16,
155
+ "prefix": null,
156
+ "problem_type": null,
157
+ "pruned_heads": {},
158
+ "qkv_bias": true,
159
+ "remove_invalid_values": false,
160
+ "repetition_penalty": 1.0,
161
+ "return_dict": true,
162
+ "return_dict_in_generate": false,
163
+ "sep_token_id": null,
164
+ "suppress_tokens": null,
165
+ "task_specific_params": null,
166
+ "temperature": 1.0,
167
+ "tf_legacy_loss": false,
168
+ "tie_encoder_decoder": false,
169
+ "tie_word_embeddings": true,
170
+ "tokenizer_class": null,
171
+ "top_k": 50,
172
+ "top_p": 1.0,
173
+ "torch_dtype": null,
174
+ "torchscript": false,
175
+ "typical_p": 1.0,
176
+ "use_bfloat16": false
177
+ },
178
+ "eos_token_id": 50265,
179
+ "is_encoder_decoder": true,
180
+ "model_type": "vision-encoder-decoder",
181
+ "pad_token_id": 50265,
182
+ "tie_word_embeddings": false,
183
+ "torch_dtype": "float32",
184
+ "transformers_version": "4.38.2"
185
+ }
generation_config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 50256,
3
+ "eos_token_id": 50256,
4
+ "transformers_version": "4.38.2"
5
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cba038b7444110495a083c76bc88080dd0f8f0596799003b1e2572622981f53f
3
+ size 956869312
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTFeatureExtractor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
runs/Apr20_12-36-50_f90af403cfaa/events.out.tfevents.1713616613.f90af403cfaa.4362.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3562bbfb0609a5fc68cf2826af1d8a7df86eb26b4663e4c7ebc22fca61420c95
3
+ size 9177
runs/Apr20_12-44-48_f90af403cfaa/events.out.tfevents.1713617100.f90af403cfaa.4362.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:808e5772ec38b3d8c0d25b7023c4707a26d8c8f6c75a1b5388a16f7e8b597ce0
3
+ size 8882
runs/Apr20_12-44-48_f90af403cfaa/events.out.tfevents.1713617114.f90af403cfaa.4362.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d590ac739c314918ff71785411704696fe5ead17966281d6e1c16131faf5d995
3
+ size 11318
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49562acc3c50c182c50dd1afec2d89d8ef4a19d23c99ad67c45fa9b848a909da
3
+ size 5048