dtd-ConvNeXt-base / config.json
anonauthors's picture
Upload ConvNextForImageClassification
03eea2b verified
raw
history blame
2.64 kB
{
"_name_or_path": "facebook/convnext-base-224-22k-1k",
"architectures": [
"ConvNextForImageClassification"
],
"depths": [
3,
3,
27,
3
],
"drop_path_rate": 0.0,
"hidden_act": "gelu",
"hidden_sizes": [
128,
256,
512,
1024
],
"id2label": {
"0": "banded",
"1": "blotchy",
"2": "braided",
"3": "bubbly",
"4": "bumpy",
"5": "chequered",
"6": "cobwebbed",
"7": "cracked",
"8": "crosshatched",
"9": "crystalline",
"10": "dotted",
"11": "fibrous",
"12": "flecked",
"13": "freckled",
"14": "frilly",
"15": "gauzy",
"16": "grid",
"17": "grooved",
"18": "honeycombed",
"19": "interlaced",
"20": "knitted",
"21": "lacelike",
"22": "lined",
"23": "marbled",
"24": "matted",
"25": "meshed",
"26": "paisley",
"27": "perforated",
"28": "pitted",
"29": "pleated",
"30": "polka-dotted",
"31": "porous",
"32": "potholed",
"33": "scaly",
"34": "smeared",
"35": "spiralled",
"36": "sprinkled",
"37": "stained",
"38": "stratified",
"39": "striped",
"40": "studded",
"41": "swirly",
"42": "veined",
"43": "waffled",
"44": "woven",
"45": "wrinkled",
"46": "zigzagged"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"banded": 0,
"blotchy": 1,
"braided": 2,
"bubbly": 3,
"bumpy": 4,
"chequered": 5,
"cobwebbed": 6,
"cracked": 7,
"crosshatched": 8,
"crystalline": 9,
"dotted": 10,
"fibrous": 11,
"flecked": 12,
"freckled": 13,
"frilly": 14,
"gauzy": 15,
"grid": 16,
"grooved": 17,
"honeycombed": 18,
"interlaced": 19,
"knitted": 20,
"lacelike": 21,
"lined": 22,
"marbled": 23,
"matted": 24,
"meshed": 25,
"paisley": 26,
"perforated": 27,
"pitted": 28,
"pleated": 29,
"polka-dotted": 30,
"porous": 31,
"potholed": 32,
"scaly": 33,
"smeared": 34,
"spiralled": 35,
"sprinkled": 36,
"stained": 37,
"stratified": 38,
"striped": 39,
"studded": 40,
"swirly": 41,
"veined": 42,
"waffled": 43,
"woven": 44,
"wrinkled": 45,
"zigzagged": 46
},
"layer_norm_eps": 1e-12,
"layer_scale_init_value": 1e-06,
"model_type": "convnext",
"num_channels": 3,
"num_stages": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.38.2"
}