| { | |
| "architecture": "resnet18", | |
| "num_classes": 4, | |
| "num_features": 512, | |
| "label_names": [ | |
| "a", | |
| "b", | |
| "c", | |
| "d" | |
| ], | |
| "pretrained_cfg": { | |
| "custom_load": false, | |
| "input_size": [ | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "fixed_input_size": false, | |
| "interpolation": "bilinear", | |
| "crop_pct": 0.875, | |
| "crop_mode": "center", | |
| "mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "num_classes": 1000, | |
| "pool_size": [ | |
| 7, | |
| 7 | |
| ], | |
| "first_conv": "conv1", | |
| "classifier": "fc" | |
| } | |
| } |