vit-base-beans / config.json
amyeroberts's picture
amyeroberts HF staff
Training in progress, epoch 1
c14adcb verified
{
"_name_or_path": "timm/resnet18.a1_in1k",
"architecture": "resnet18",
"architectures": [
"TimmWrapperForImageClassification"
],
"finetuning_task": "image-classification",
"id2label": {
"0": "angular_leaf_spot",
"1": "bean_rust",
"2": "healthy"
},
"initializer_range": 0.02,
"label2id": {
"angular_leaf_spot": "0",
"bean_rust": "1",
"healthy": "2"
},
"model_name": "timm/resnet18.a1_in1k",
"model_type": "timm_wrapper",
"num_classes": 1000,
"num_features": 512,
"pretrained_cfg": {
"classifier": "fc",
"crop_mode": "center",
"crop_pct": 0.95,
"custom_load": false,
"first_conv": "conv1",
"fixed_input_size": false,
"input_size": [
3,
224,
224
],
"interpolation": "bicubic",
"mean": [
0.485,
0.456,
0.406
],
"num_classes": 1000,
"origin_url": "https://github.com/huggingface/pytorch-image-models",
"paper_ids": "arXiv:2110.00476",
"pool_size": [
7,
7
],
"std": [
0.229,
0.224,
0.225
],
"tag": "a1_in1k",
"test_crop_pct": 1.0,
"test_input_size": [
3,
288,
288
]
},
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.46.0.dev0"
}