{
  "source_run": "https://wandb.ai/tsilva/dlab/runs/fs4fa4vl",
  "test_run": "https://wandb.ai/tsilva/dlab/runs/enmuxlt3",
  "source_artifact": "https://wandb.ai/tsilva/dlab/artifacts/run-output/mnist-rnn_001-rnn-architecture-search_adamw-lr0p003-bs512-cosine_w384-d1-do0p2_val-acc-confirm-top-i000_seed9001-run/latest",
  "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
    ]
  },
  "architecture": {
    "name": "rnn",
    "input_size": null,
    "hidden_dim": 384,
    "num_layers": 1,
    "dropout": 0.2,
    "rnn_type": "gru",
    "bidirectional": false,
    "pooling": "last",
    "sequence_axis": "columns"
  },
  "metrics": {
    "selected_validation_accuracy": 0.994667,
    "selected_validation_loss": 0.018941,
    "test_accuracy": 0.9917,
    "test_loss": 0.027544,
    "three_seed_validation_accuracy_mean": 0.993667,
    "three_seed_validation_accuracy_std": 0.0008164965809277268,
    "three_seed_test_accuracy_mean": 0.9927,
    "three_seed_test_accuracy_std": 0.0007483314773547949
  }
}