| { | |
| "architectures": [ | |
| "RSPResNetForImageClassification" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_rsp.RSPResNetConfig", | |
| "AutoModelForImageClassification": "modeling_rsp.RSPResNetForImageClassification" | |
| }, | |
| "block": "Bottleneck", | |
| "image_size": 224, | |
| "layers": [ | |
| 3, | |
| 4, | |
| 6, | |
| 3 | |
| ], | |
| "model_type": "rsp_resnet", | |
| "num_channels": 3, | |
| "num_labels": 51, | |
| "source_checkpoint": "/mnt/data/projects/model_hubs/raw/rsp-resnet-50-ckpt.pth" | |
| } |