zm_config.json

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.1",
  "model_module": "zeromodels.models.segformer",
  "model_class": "SegFormerSemanticSegment",
  "variant": "segformer_b3_cityscapes_1024",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "segformer",
  "vision_config": {
    "embed_dim": [
      64,
      128,
      320,
      512
    ],
    "depths": [
      3,
      4,
      18,
      3
    ],
    "decode_head_dim": 768,
    "dropout_rate": 0.1,
    "num_classes": 19,
    "image_size": 1024
  }
}