prism-seek / config.json
Croznut's picture
Training in progress, epoch 1
d4420b3 verified
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"dtype": "float32",
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "badger_scat",
"1": "bear_scat",
"2": "beaver_scat",
"3": "bison_scat",
"4": "black_bear_scat",
"5": "bobcat_scat",
"6": "cat_scat",
"7": "cow_scat",
"8": "coyote_scat",
"9": "deer_scat",
"10": "dog_scat",
"11": "elk_scat",
"12": "fox_scat",
"13": "goat_scat",
"14": "grizzly_bear_scat",
"15": "horse_scat",
"16": "lynx_scat",
"17": "moose_scat",
"18": "mountain_lion_scat",
"19": "mule_deer_scat",
"20": "opossum_scat",
"21": "otter_scat",
"22": "pig_scat",
"23": "porcupine_scat",
"24": "pronghorn_scat",
"25": "rabbit_scat",
"26": "raccoon_scat",
"27": "sheep_scat",
"28": "skunk_scat",
"29": "squirrel_scat",
"30": "wolf_scat"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"badger_scat": 0,
"bear_scat": 1,
"beaver_scat": 2,
"bison_scat": 3,
"black_bear_scat": 4,
"bobcat_scat": 5,
"cat_scat": 6,
"cow_scat": 7,
"coyote_scat": 8,
"deer_scat": 9,
"dog_scat": 10,
"elk_scat": 11,
"fox_scat": 12,
"goat_scat": 13,
"grizzly_bear_scat": 14,
"horse_scat": 15,
"lynx_scat": 16,
"moose_scat": 17,
"mountain_lion_scat": 18,
"mule_deer_scat": 19,
"opossum_scat": 20,
"otter_scat": 21,
"pig_scat": 22,
"porcupine_scat": 23,
"pronghorn_scat": 24,
"rabbit_scat": 25,
"raccoon_scat": 26,
"sheep_scat": 27,
"skunk_scat": 28,
"squirrel_scat": 29,
"wolf_scat": 30
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"pooler_act": "tanh",
"pooler_output_size": 768,
"problem_type": "single_label_classification",
"qkv_bias": true,
"transformers_version": "4.57.0"
}