ZafarLocAI's picture
Model save
c353b17 verified
{
"architectures": [
"ConvNextV2ForImageClassification"
],
"depths": [
3,
3,
27,
3
],
"drop_path_rate": 0.0,
"dtype": "float32",
"hidden_act": "gelu",
"hidden_sizes": [
192,
384,
768,
1536
],
"id2label": {
"0": "A1",
"1": "A10",
"2": "A11",
"3": "A12",
"4": "A13",
"5": "A14",
"6": "A15",
"7": "A16",
"8": "A17",
"9": "A18",
"10": "A19",
"11": "A2",
"12": "A20",
"13": "A3",
"14": "A4",
"15": "A5",
"16": "A6",
"17": "A7",
"18": "A8",
"19": "A9"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"A1": 0,
"A10": 1,
"A11": 2,
"A12": 3,
"A13": 4,
"A14": 5,
"A15": 6,
"A16": 7,
"A17": 8,
"A18": 9,
"A19": 10,
"A2": 11,
"A20": 12,
"A3": 13,
"A4": 14,
"A5": 15,
"A6": 16,
"A7": 17,
"A8": 18,
"A9": 19
},
"layer_norm_eps": 1e-12,
"model_type": "convnextv2",
"num_channels": 3,
"num_stages": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"transformers_version": "5.3.0",
"use_cache": false
}