Training in progress, epoch 1
#3
by
LeeTung
- opened
- config.json +7 -3
- generation_config.json +1 -0
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -22,7 +22,6 @@
|
|
| 22 |
"init_std": 0.02,
|
| 23 |
"is_decoder": true,
|
| 24 |
"is_encoder_decoder": false,
|
| 25 |
-
"max_length": 768,
|
| 26 |
"max_position_embeddings": 1536,
|
| 27 |
"model_type": "mbart",
|
| 28 |
"num_hidden_layers": 12,
|
|
@@ -46,8 +45,8 @@
|
|
| 46 |
"hidden_dropout_prob": 0.0,
|
| 47 |
"hidden_size": 1024,
|
| 48 |
"image_size": [
|
| 49 |
-
|
| 50 |
-
|
| 51 |
],
|
| 52 |
"initializer_range": 0.02,
|
| 53 |
"layer_norm_eps": 1e-05,
|
|
@@ -68,7 +67,12 @@
|
|
| 68 |
"use_absolute_embeddings": false,
|
| 69 |
"window_size": 10
|
| 70 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
"is_encoder_decoder": true,
|
|
|
|
| 72 |
"model_type": "vision-encoder-decoder",
|
| 73 |
"pad_token_id": 1,
|
| 74 |
"tie_word_embeddings": false,
|
|
|
|
| 22 |
"init_std": 0.02,
|
| 23 |
"is_decoder": true,
|
| 24 |
"is_encoder_decoder": false,
|
|
|
|
| 25 |
"max_position_embeddings": 1536,
|
| 26 |
"model_type": "mbart",
|
| 27 |
"num_hidden_layers": 12,
|
|
|
|
| 45 |
"hidden_dropout_prob": 0.0,
|
| 46 |
"hidden_size": 1024,
|
| 47 |
"image_size": [
|
| 48 |
+
2560,
|
| 49 |
+
1920
|
| 50 |
],
|
| 51 |
"initializer_range": 0.02,
|
| 52 |
"layer_norm_eps": 1e-05,
|
|
|
|
| 67 |
"use_absolute_embeddings": false,
|
| 68 |
"window_size": 10
|
| 69 |
},
|
| 70 |
+
"image_size": [
|
| 71 |
+
1280,
|
| 72 |
+
960
|
| 73 |
+
],
|
| 74 |
"is_encoder_decoder": true,
|
| 75 |
+
"max_length": null,
|
| 76 |
"model_type": "vision-encoder-decoder",
|
| 77 |
"pad_token_id": 1,
|
| 78 |
"tie_word_embeddings": false,
|
generation_config.json
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 0,
|
|
|
|
| 4 |
"eos_token_id": 2,
|
| 5 |
"forced_eos_token_id": 2,
|
| 6 |
"max_length": 768,
|
|
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 0,
|
| 4 |
+
"decoder_start_token_id": 57579,
|
| 5 |
"eos_token_id": 2,
|
| 6 |
"forced_eos_token_id": 2,
|
| 7 |
"max_length": 768,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 809296024
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:65085b2355127df2877f3dd6a176241aa5a5fd6152c942697546d33bf64863cc
|
| 3 |
size 809296024
|