vpr/config.json

{
  "architectures": [
    "AnyThermalVPRModel"
  ],
  "model_type": "anythermal_vpr",
  "auto_map": {
    "AutoConfig": "model.AnyThermalVPRConfig",
    "AutoModel": "model.AnyThermalVPRModel"
  },
  "hidden_size": 768,
  "num_hidden_layers": 12,
  "num_attention_heads": 12,
  "mlp_ratio": 4,
  "hidden_act": "gelu",
  "dropout": 0.0,
  "attention_dropout": 0.0,
  "initializer_range": 0.02,
  "layer_norm_eps": 1e-06,
  "image_size": 518,
  "patch_size": 14,
  "num_channels": 3,
  "qkv_bias": true,
  "layerscale_value": 1.0,
  "drop_path_rate": 0.0,
  "use_swiglu_ffn": false,
  "num_clusters": 64,
  "cluster_dim": 128,
  "token_dim": 256
}