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