{
    "architectures": ["MiniVisionV2"],
    "model_type": "minivision_v2",
    "framework": "PyTorch",
    "task": "image-classification",
    "num_labels": 10,
    "input_channels": 1,
    "image_size": 28,

    "model_config": {
        "conv1_out_channels": 32,
        "conv2_out_channels": 64,
        "hidden_size": 256,
        "dropout_rate": 0.3,
        "kernel_size": 3
    },

    "training_config": {
        "dataset": "ylecun/mnist",
        "batch_size": 256,
        "learning_rate": 0.01,
        "optimizer": "SGD",
        "momentum": 0.8,
        "scheduler": "StepLR",
        "scheduler_step_size": 3,
        "scheduler_gamma": 0.5,
        "epochs": 50
    },

    "id2label": {
        "0": "0",
        "1": "1",
        "2": "2",
        "3": "3",
        "4": "4",
        "5": "5",
        "6": "6",
        "7": "7",
        "8": "8",
        "9": "9"
    },

    "label2id": {
        "0": "0",
        "1": "1",
        "2": "2",
        "3": "3",
        "4": "4",
        "5": "5",
        "6": "6",
        "7": "7",
        "8": "8",
        "9": "9"
    }
}