Commit
·
33ff3a3
1
Parent(s):
63ddd96
Initial commit for new model
Browse files- config.json +25 -0
config.json
CHANGED
@@ -250,5 +250,30 @@
|
|
250 |
"torchscript": false,
|
251 |
"typical_p": 1.0,
|
252 |
"use_bfloat16": false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
253 |
}
|
254 |
}
|
|
|
250 |
"torchscript": false,
|
251 |
"typical_p": 1.0,
|
252 |
"use_bfloat16": false
|
253 |
+
},
|
254 |
+
"spatial_encoder_hidden_size": 768,
|
255 |
+
"num_attention_heads": 16,
|
256 |
+
"spatial_encoder_cfg": {
|
257 |
+
"architectures": [
|
258 |
+
"Mast3rEncoder"
|
259 |
+
],
|
260 |
+
"pretrained_model_name_or_path": "../mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth",
|
261 |
+
"preprocessor_config": {
|
262 |
+
"do_convert_rgb": true,
|
263 |
+
"do_normalize": true,
|
264 |
+
"do_rescale": true,
|
265 |
+
"do_resize_and_center_crop": true,
|
266 |
+
"image_mean": [0.5, 0.5, 0.5],
|
267 |
+
"image_std": [0.5, 0.5, 0.5],
|
268 |
+
"resample": 3,
|
269 |
+
"rescale_factor": 0.00392156862745098,
|
270 |
+
"longest_edge": 512
|
271 |
+
}
|
272 |
+
},
|
273 |
+
"cross_attention_cfg": {
|
274 |
+
"architectures": [
|
275 |
+
"CrossAttention"
|
276 |
+
],
|
277 |
+
"cross_attention_type": "cross_attn"
|
278 |
}
|
279 |
}
|