mnist-classifier-gru / metadata.json
tsilva's picture
Add MNIST GRU classifier
0ba986b verified
Raw
History Blame Contribute Delete
1.34 kB
{
"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
}
}