| { | |
| "architecture": "deit-base-patch16-224", | |
| "num_classes": 4, | |
| "num_features": 1024, | |
| "global_pool": "avg", | |
| "label_names": ["Optimistic", "Pessimistic", "Hostile", "Neutral"], | |
| "pretrained_cfg": { | |
| "tag": "custom", | |
| "custom_load": false, | |
| "input_size": [3, 224, 224], | |
| "fixed_input_size": true, | |
| "interpolation": "bicubic", | |
| "crop_pct": 0.9, | |
| "crop_mode": "center", | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225], | |
| "num_classes": 4, | |
| "pool_size": [7, 7], | |
| "first_conv": "patch_embed.proj", | |
| "classifier": "head.fc" | |
| } | |
| } | |