config.json

{
  "architecture": "LargeNet",
  "num_classes": 7,
  "input_size": [
    128,
    128
  ],
  "input_channels": 3,
  "class_names": [
    "battery",
    "biological",
    "cardboard",
    "glass",
    "metal",
    "paper",
    "plastic"
  ],
  "normalization": {
    "mean": [
      0.5,
      0.5,
      0.5
    ],
    "std": [
      0.5,
      0.5,
      0.5
    ]
  },
  "training_params": {
    "batch_size": 64,
    "learning_rate": 0.01,
    "best_epoch": 61,
    "optimizer": "SGD",
    "loss_function": "CrossEntropyLoss"
  }
}