hf-transformers-bot's picture
Update tiny models for EfficientLoFTRForKeypointMatching
e6fd10a verified
Raw
History Blame Contribute Delete
1.45 kB
{
"activation_function": "relu",
"architectures": [
"EfficientLoFTRForKeypointMatching"
],
"attention_bias": false,
"attention_dropout": 0.0,
"batch_norm_eps": 1e-05,
"coarse_matching_border_removal": 0,
"coarse_matching_skip_softmax": false,
"coarse_matching_temperature": 0.1,
"coarse_matching_threshold": 0.0,
"dtype": "float32",
"fine_fusion_dims": [
16
],
"fine_kernel_size": 2,
"fine_matching_regress_temperature": 10.0,
"fine_matching_slice_dim": 8,
"hidden_size": 16,
"initializer_range": 0.02,
"intermediate_size": 32,
"kv_aggregation_kernel_size": 1,
"kv_aggregation_stride": 1,
"mlp_activation_function": "leaky_relu",
"model_type": "efficientloftr",
"num_attention_heads": 8,
"num_attention_layers": 2,
"num_key_value_heads": 8,
"out_features": [
16,
16
],
"partial_rotary_factor": 4.0,
"q_aggregation_kernel_size": 1,
"q_aggregation_stride": 1,
"rope_parameters": {
"partial_rotary_factor": 4.0,
"rope_theta": 10000.0,
"rope_type": "default"
},
"stage_block_in_channels": [
[
1
],
[
16
]
],
"stage_block_out_channels": [
[
16
],
[
16
]
],
"stage_block_stride": [
[
2
],
[
1
]
],
"stage_in_channels": [
1,
16
],
"stage_num_blocks": [
1,
1
],
"stage_stride": [
2,
1
],
"transformers_version": "5.16.0.dev0"
}