{
  "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"
}