zm_config.json

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.6",
  "model_module": "zeromodels.models.pvt_v2",
  "model_class": "PvtV2ImageClassify",
  "variant": "pvt-v2-b0",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "model_type": "pvt_v2",
  "vision_config": {
    "hidden_sizes": [
      32,
      64,
      160,
      256
    ],
    "depths": [
      2,
      2,
      2,
      2
    ],
    "num_attention_heads": [
      1,
      2,
      5,
      8
    ],
    "sr_ratios": [
      8,
      4,
      2,
      1
    ],
    "mlp_ratios": [
      8,
      8,
      4,
      4
    ],
    "linear_attention": false,
    "image_size": 224,
    "num_classes": 1000
  }
}