Training in progress, epoch 1
#6
by
fahmiaziz
- opened
- config.json +9 -4
- generation_config.json +1 -1
- pytorch_model.bin +1 -1
config.json
CHANGED
@@ -49,7 +49,7 @@
|
|
49 |
"LABEL_1": 1
|
50 |
},
|
51 |
"length_penalty": 1.0,
|
52 |
-
"max_length":
|
53 |
"max_position_embeddings": 1536,
|
54 |
"min_length": 0,
|
55 |
"model_type": "mbart",
|
@@ -124,8 +124,8 @@
|
|
124 |
"1": "LABEL_1"
|
125 |
},
|
126 |
"image_size": [
|
127 |
-
|
128 |
-
|
129 |
],
|
130 |
"initializer_range": 0.02,
|
131 |
"is_decoder": false,
|
@@ -183,10 +183,15 @@
|
|
183 |
"use_bfloat16": false,
|
184 |
"window_size": 10
|
185 |
},
|
|
|
|
|
|
|
|
|
186 |
"is_encoder_decoder": true,
|
|
|
187 |
"model_type": "vision-encoder-decoder",
|
188 |
"pad_token_id": 1,
|
189 |
"tie_word_embeddings": false,
|
190 |
"torch_dtype": "float32",
|
191 |
-
"transformers_version": "4.
|
192 |
}
|
|
|
49 |
"LABEL_1": 1
|
50 |
},
|
51 |
"length_penalty": 1.0,
|
52 |
+
"max_length": 20,
|
53 |
"max_position_embeddings": 1536,
|
54 |
"min_length": 0,
|
55 |
"model_type": "mbart",
|
|
|
124 |
"1": "LABEL_1"
|
125 |
},
|
126 |
"image_size": [
|
127 |
+
2560,
|
128 |
+
1920
|
129 |
],
|
130 |
"initializer_range": 0.02,
|
131 |
"is_decoder": false,
|
|
|
183 |
"use_bfloat16": false,
|
184 |
"window_size": 10
|
185 |
},
|
186 |
+
"image_size": [
|
187 |
+
1280,
|
188 |
+
960
|
189 |
+
],
|
190 |
"is_encoder_decoder": true,
|
191 |
+
"max_length": 768,
|
192 |
"model_type": "vision-encoder-decoder",
|
193 |
"pad_token_id": 1,
|
194 |
"tie_word_embeddings": false,
|
195 |
"torch_dtype": "float32",
|
196 |
+
"transformers_version": "4.33.1"
|
197 |
}
|
generation_config.json
CHANGED
@@ -6,5 +6,5 @@
|
|
6 |
"forced_eos_token_id": 2,
|
7 |
"max_length": 768,
|
8 |
"pad_token_id": 1,
|
9 |
-
"transformers_version": "4.
|
10 |
}
|
|
|
6 |
"forced_eos_token_id": 2,
|
7 |
"max_length": 768,
|
8 |
"pad_token_id": 1,
|
9 |
+
"transformers_version": "4.33.1"
|
10 |
}
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 809404185
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:51b0e04fad1933c6aac8e7de0246f9d4cbad77869064354cbd1102f77d7c5924
|
3 |
size 809404185
|