matchanything_eloftr / config.json
stevenbucaille's picture
Add model
9bf358b verified
{
"activation_function": "relu",
"architectures": [
"EfficientLoFTRForKeypointMatching"
],
"attention_bias": false,
"attention_dropout": 0.0,
"batch_norm_eps": 1e-05,
"coarse_matching_border_removal": 2,
"coarse_matching_skip_softmax": false,
"coarse_matching_temperature": 0.1,
"coarse_matching_threshold": 0.2,
"embedding_size": [
26,
26
],
"fine_fusion_dims": [
256,
128,
64
],
"fine_kernel_size": 8,
"fine_matching_regress_temperature": 10.0,
"fine_matching_slice_dim": 8,
"hidden_size": 256,
"initializer_range": 0.02,
"intermediate_size": 512,
"kv_aggregation_kernel_size": 4,
"kv_aggregation_stride": 4,
"mlp_activation_function": "leaky_relu",
"model_type": "efficientloftr",
"num_attention_heads": 8,
"num_attention_layers": 4,
"num_key_value_heads": 8,
"out_features": [
64,
64,
128,
256
],
"partial_rotary_factor": 4.0,
"q_aggregation_kernel_size": 4,
"q_aggregation_stride": 4,
"rope_scaling": {
"rope_type": "default"
},
"rope_theta": 10000.0,
"stage_block_in_channels": [
[
1
],
[
64,
64
],
[
64,
128,
128,
128
],
[
128,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256
]
],
"stage_block_out_channels": [
[
64
],
[
64,
64
],
[
128,
128,
128,
128
],
[
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256,
256
]
],
"stage_block_stride": [
[
2
],
[
1,
1
],
[
2,
1,
1,
1
],
[
2,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1
]
],
"stage_in_channels": [
1,
64,
64,
128
],
"stage_num_blocks": [
1,
2,
4,
14
],
"stage_stride": [
2,
1,
2,
2
],
"torch_dtype": "float32",
"transformers_version": "4.56.0.dev0"
}