| { | |
| "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" | |
| } | |
| } |