| { | |
| "architecture": "convformer_s18", | |
| "num_classes": 1000, | |
| "num_features": 512, | |
| "pretrained_cfg": { | |
| "tag": "st_safebooru_1k", | |
| "custom_load": true, | |
| "input_size": [ | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "fixed_input_size": false, | |
| "interpolation": "bicubic", | |
| "crop_pct": 1.0, | |
| "crop_mode": "center", | |
| "mean": [ | |
| 0.6886, | |
| 0.6415, | |
| 0.6443 | |
| ], | |
| "std": [ | |
| 0.183, | |
| 0.1852, | |
| 0.1849 | |
| ], | |
| "num_classes": 1000, | |
| "pool_size": [ | |
| 7, | |
| 7 | |
| ], | |
| "first_conv": "stem.conv", | |
| "classifier": "head.fc.fc2" | |
| } | |
| } |