SI-IMAGE-2 / config.json
eyad-silx's picture
Update config.json
04500ea verified
{
"_name_or_path": "densenet-finetuned",
"architectures": [
"DenseNetForImageClassification"
],
"model_type": "image-classification",
"num_labels": 3,
"hidden_size": 1024,
"label2id": {
"Real": 0,
"AI": 1,
"Other": 2
},
"id2label": {
"0": "Real",
"1": "AI",
"2": "Other"
},
"torchvision_backbone": "densenet121",
"fine_tuned": true,
"framework": "pytorch"
}