{
  "source_run": "https://wandb.ai/tsilva/dlab/runs/652i33os",
  "test_run": "https://wandb.ai/tsilva/dlab/runs/ntzpryf9",
  "source_artifact": "https://wandb.ai/tsilva/dlab/artifacts/run-output/mnist-rnn_001-rnn-architecture-search_adamw-lr0p003-bs512-cosine_w384-d2-do0p1-ls0p01_val-acc-confirm-top-i001_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": 2,
    "dropout": 0.1,
    "rnn_type": "lstm",
    "bidirectional": false,
    "pooling": "last",
    "sequence_axis": "columns"
  },
  "metrics": {
    "selected_validation_accuracy": 0.993833,
    "selected_validation_loss": 0.091574,
    "test_accuracy": 0.9931,
    "test_loss": 0.092943,
    "three_seed_validation_accuracy_mean": 0.9925,
    "three_seed_validation_accuracy_std": 0.0011625801764466143,
    "three_seed_test_accuracy_mean": 0.9924333333333334,
    "three_seed_test_accuracy_std": 0.0005734883511361592
  }
}