| { | |
| "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": [ | |
| 15, | |
| 20 | |
| ], | |
| "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.54.0.dev0" | |
| } | |