config.json

{
  "architectures": [
    "CVLFaceRecognitionModel"
  ],
  "auto_map": {
    "AutoConfig": "wrapper.ModelConfig",
    "AutoModel": "wrapper.CVLFaceRecognitionModel"
  },
  "conf": {
    "color_space": "RGB",
    "freeze": false,
    "input_size": [
      3,
      112,
      112
    ],
    "mask_ratio": 0.0,
    "name": "base",
    "output_dim": 512,
    "rpe_config": {
      "ctx_type": "rel_keypoint_splithead_unshared",
      "method": "product",
      "mode": "ctx",
      "name": "KPRPE_shared",
      "num_keypoints": 5,
      "ratio": 1.9,
      "rpe_on": "k",
      "shared_head": true
    },
    "start_from": "",
    "yaml_path": "models/vit_kprpe/configs/v1_base_kprpe_splithead_unshared.yaml"
  },
  "torch_dtype": "float32",
  "transformers_version": "4.33.0"
}