resnet-imagenet1k / config.json
nragrawal's picture
Update model name
1efe7e3
raw
history blame contribute delete
591 Bytes
{
"architectures": [
"ResNet50"
],
"model_type": "resnet",
"num_channels": 3,
"num_classes": 1000,
"image_size": 224,
"hidden_sizes": [64, 128, 256, 512],
"preprocessing": {
"image_mean": [0.485, 0.456, 0.406],
"image_std": [0.229, 0.224, 0.225],
"size": {
"height": 224,
"width": 224
}
},
"id2label": {
"0": "tench",
"1": "goldfish",
"999": "toilet_tissue"
},
"label2id": {
"tench": 0,
"goldfish": 1,
"toilet_tissue": 999
}
}