{
    "architecture": "Custom CNN",
    "input_size": [
        224,
        224
    ],
    "num_classes": 6,
    "transformations": [
        "RandomHorizontalFlip",
        "RandomRotation(15)",
        "ColorJitter(brightness=0.2, contrast=0.2)",
        "Resize(224x224)",
        "ToTensor",
        "Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])"
    ],
    "optimizer": "Adam",
    "learning_rate": 0.001,
    "batch_size": 32,
    "epochs": 10
}