config.json

{
  "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,
  "fine_kernel_size": 8,
  "fine_matching_regress_temperature": 10.0,
  "fine_matching_slice_dim": 8,
  "hidden_size": 256,
  "initializer_range": 0.02,
  "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,
  "q_aggregation_kernel_size": 4,
  "q_aggregation_stride": 4,
  "rope_scaling": {
    "dim": 64,
    "rope_type": "2d"
  },
  "rope_theta": 10000.0,
  "stage_block_dims": [
    64,
    64,
    128,
    256
  ],
  "stage_hidden_expansion": [
    1,
    1,
    1,
    1
  ],
  "stage_num_blocks": [
    1,
    2,
    4,
    14
  ],
  "stage_stride": [
    2,
    1,
    2,
    2
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.50.0.dev0"
}