| { | |
| "model_name": "mnist-mlp", | |
| "architecture": "MLP", | |
| "task": "image-classification", | |
| "dataset": "MNIST", | |
| "framework": "PyTorch", | |
| "input_size": 784, | |
| "hidden_layers": [128, 64], | |
| "activation": "relu", | |
| "output_size": 10, | |
| "loss_function": "cross_entropy", | |
| "optimizer": "adam", | |
| "image_size": [28, 28], | |
| "num_classes": 10 | |
| } | |