| { | |
| "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 | |
| } | |
| } | |