metadata.json

{
  "source_run": "https://wandb.ai/tsilva/dlab/runs/gsuy1ifx",
  "source_sweep": "https://wandb.ai/tsilva/dlab/sweeps/xa56lubb",
  "checkpoint": "model.ckpt",
  "onnx": "model.onnx",
  "input_shape": [
    1,
    28,
    28
  ],
  "input_name": "images",
  "output_name": "logits",
  "classes": [
    "0",
    "1",
    "2",
    "3",
    "4",
    "5",
    "6",
    "7",
    "8",
    "9"
  ],
  "preprocessing": {
    "to_tensor": true,
    "normalize_mean": [
      0.1307
    ],
    "normalize_std": [
      0.3081
    ]
  },
  "metrics": {
    "validation_accuracy": 0.9950000047683716,
    "test_accuracy": 0.9947999715805054,
    "test_loss": 0.14728698134422302,
    "test_errors": 52
  }
}