zm_config.json

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.7",
  "model_module": "zeromodels.models.regnet",
  "model_class": "RegNetImageClassify",
  "variant": "regnet-y-320",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "model_type": "regnet",
  "vision_config": {
    "embedding_size": 32,
    "hidden_sizes": [
      232,
      696,
      1392,
      3712
    ],
    "depths": [
      2,
      5,
      12,
      1
    ],
    "groups_width": 232,
    "layer_type": "y",
    "downsample_in_first_stage": true,
    "num_classes": 1000
  }
}