| { | |
| "model_type": "resnet50", | |
| "task": "image-classification", | |
| "num_labels": 1, | |
| "id2label": {"0": "cat", "1": "dog"}, | |
| "image_size": 224, | |
| "normalization": { | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225] | |
| }, | |
| "backbone": "torchvision resnet50 (IMAGENET1K_V1)", | |
| "head": "Dropout(0.4) -> Linear(2048, 1)", | |
| "output": "single logit; sigmoid(logit) > 0.5 => dog", | |
| "weights_file": "cat_dog_classifier.pt", | |
| "library_name": "pytorch", | |
| "license": "mit" | |
| } | |