config.json

{
  "model_type": "resnet18",
  "num_classes": 10,
  "class_names": [
    "battery",
    "biological",
    "cardboard",
    "clothes",
    "glass",
    "metal",
    "paper",
    "plastic",
    "shoes",
    "trash"
  ],
  "task": "image-classification",
  "dataset": "Garbage Classification v2 (Kaggle)"
}