levit-256 / zm_config.json

Upload folder using huggingface_hub

081accb verified 2 months ago

847 Bytes

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.7",
  "model_module": "zeromodels.models.levit",
  "model_class": "LevitImageClassify",
  "variant": "levit-256",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "model_type": "levit",
  "vision_config": {
    "image_size": 224,
    "num_channels": 3,
    "kernel_size": 3,
    "stride": 2,
    "padding": 1,
    "patch_size": 16,
    "hidden_sizes": [
      256,
      384,
      512
    ],
    "num_attention_heads": [
      4,
      6,
      8
    ],
    "depths": [
      4,
      4,
      4
    ],
    "key_dim": [
      32,
      32,
      32
    ],
    "mlp_ratio": [
      2,
      2,
      2
    ],
    "attention_ratio": [
      2,
      2,
      2
    ],
    "num_classes": 1000,
    "use_distillation": true
  }
}

	{
	"library_name": "zeromodels",
	"zeromodels_version": "1.2.7",
	"model_module": "zeromodels.models.levit",
	"model_class": "LevitImageClassify",
	"variant": "levit-256",
	"weights": "model.weights.h5",
	"schema_version": 2,
	"model_type": "levit",
	"vision_config": {
	"image_size": 224,
	"num_channels": 3,
	"kernel_size": 3,
	"stride": 2,
	"padding": 1,
	"patch_size": 16,
	"hidden_sizes": [
	256,
	384,
	512
	],
	"num_attention_heads": [
	4,
	6,
	8
	],
	"depths": [
	4,
	4,
	4
	],
	"key_dim": [
	32,
	32,
	32
	],
	"mlp_ratio": [
	2,
	2,
	2
	],
	"attention_ratio": [
	2,
	2,
	2
	],
	"num_classes": 1000,
	"use_distillation": true
	}
	}