mnist-classifier-rnn / metadata.json
tsilva's picture
Add MNIST RNN classifier
620e95c verified
Raw
History Blame Contribute Delete
1.36 kB
{
"source_run": "https://wandb.ai/tsilva/dlab/runs/t8u8llqo",
"test_run": "https://wandb.ai/tsilva/dlab/runs/vdt5duxq",
"source_artifact": "https://wandb.ai/tsilva/dlab/artifacts/run-output/mnist-rnn_001-rnn-architecture-search_adamw-lr0p003-bs256-constant_w512-d2-do0_val-acc-confirm-top-i005_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": 512,
"num_layers": 2,
"dropout": 0,
"rnn_type": "rnn",
"bidirectional": false,
"pooling": "mean",
"sequence_axis": "rows"
},
"metrics": {
"selected_validation_accuracy": 0.979667,
"selected_validation_loss": 0.067957,
"test_accuracy": 0.9778,
"test_loss": 0.07622,
"three_seed_validation_accuracy_mean": 0.9791666666666666,
"three_seed_validation_accuracy_std": 0.0004907289362661005,
"three_seed_test_accuracy_mean": 0.9797333333333333,
"three_seed_test_accuracy_std": 0.0013695092389449483
}
}