config.json

{
    "model_architecture": "ResNet18",
    "num_classes": 6,
    "input_size": [224, 224],
    "pretrained": true,
    "learning_rate": 0.0001,
    "batch_size": 256
  }