training_config.json

{
  "learning_rate": 0.001,
  "hidden_units": 512,
  "epochs": 3,
  "batch_size": 32,
  "optimizer": "AdamW",
  "scheduler": {
    "type": "StepLR",
    "step_size": 5,
    "gamma": 0.1
  },
  "loss": "CrossEntropyLoss",
  "best_validation_accuracy_percent": 94.38,
  "normalization": {
    "mean": [
      0.485,
      0.456,
      0.406
    ],
    "std": [
      0.229,
      0.224,
      0.225
    ]
  },
  "train_transforms": [
    "RandomResizedCrop(224)",
    "RandomHorizontalFlip",
    "ColorJitter(brightness=0.2,contrast=0.2,saturation=0.2,hue=0.1)"
  ],
  "validation_transforms": [
    "Resize(256)",
    "CenterCrop(224)"
  ]
}