config.json

{
  "architectures": [
    "SapiensGaitModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_sapiens.SapiensGaitConfig",
    "AutoModel": "modeling_sapiens.SapiensGaitModel"
  },
  "model_type": "sapiens_gait",
  "image_size": [
    1024,
    768
  ],
  "patch_size": 16,
  "patch_padding": 2,
  "image_mean": [
    123.675,
    116.28,
    103.53
  ],
  "image_std": [
    58.395,
    57.12,
    57.375
  ],
  "in_channels": 3,
  "embed_dims": 1024,
  "num_layers": 24,
  "num_heads": 16,
  "feedforward_channels": 4096,
  "drop_rate": 0.0,
  "drop_path_rate": 0.0,
  "qkv_bias": true,
  "num_keypoints": 133,
  "deconv_out_channels": [
    768,
    768
  ],
  "deconv_kernel_sizes": [
    4,
    4
  ],
  "conv_out_channels": [
    768,
    768
  ],
  "conv_kernel_sizes": [
    1,
    1
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.40.0"
}