| { | |
| "model_name": "convnext_small", | |
| "architecture": "torchvision.models.convnext_small", | |
| "num_labels": 7, | |
| "class_names": [ | |
| "miss_cereus_repandus", | |
| "miss_myrtillocactus_geometrizans", | |
| "miss_pilosocereus_pachycladus", | |
| "trichocereus_bridgesii", | |
| "trichocereus_pachanoi", | |
| "trichocereus_peruvianus", | |
| "unknown" | |
| ], | |
| "id2label": { | |
| "0": "miss_cereus_repandus", | |
| "1": "miss_myrtillocactus_geometrizans", | |
| "2": "miss_pilosocereus_pachycladus", | |
| "3": "trichocereus_bridgesii", | |
| "4": "trichocereus_pachanoi", | |
| "5": "trichocereus_peruvianus", | |
| "6": "unknown" | |
| }, | |
| "label2id": { | |
| "miss_cereus_repandus": 0, | |
| "miss_myrtillocactus_geometrizans": 1, | |
| "miss_pilosocereus_pachycladus": 2, | |
| "trichocereus_bridgesii": 3, | |
| "trichocereus_pachanoi": 4, | |
| "trichocereus_peruvianus": 5, | |
| "unknown": 6 | |
| }, | |
| "image_size": 224, | |
| "resize_size": 256, | |
| "image_mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "image_std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "unknown_class": "unknown", | |
| "unknown_rejection_threshold": 0.0, | |
| "checkpoint_selection_metric": "macro_f1", | |
| "checkpoint_selection_score": 0.88370817372507, | |
| "weights_file": "model.pt" | |
| } | |