Bug fix image_size as dict
Browse filesModel wasnt loading when using encoder.image_size as dict, the VisionEncoderDecoderModel class expects that is an array instead of a dictionary.
- config.json +1 -4
config.json
CHANGED
@@ -123,10 +123,7 @@
|
|
123 |
"0": "LABEL_0",
|
124 |
"1": "LABEL_1"
|
125 |
},
|
126 |
-
"image_size":
|
127 |
-
"height": 2560,
|
128 |
-
"width": 1920
|
129 |
-
},
|
130 |
"initializer_range": 0.02,
|
131 |
"is_decoder": false,
|
132 |
"is_encoder_decoder": false,
|
|
|
123 |
"0": "LABEL_0",
|
124 |
"1": "LABEL_1"
|
125 |
},
|
126 |
+
"image_size": [2560, 1920],
|
|
|
|
|
|
|
127 |
"initializer_range": 0.02,
|
128 |
"is_decoder": false,
|
129 |
"is_encoder_decoder": false,
|