| { |
| "source_run": "https://wandb.ai/tsilva/dlab/runs/5um57rnu", |
| "source_sweep": "https://wandb.ai/tsilva/dlab/sweeps/ikfs5ox8", |
| "selection_rule": "Uploaded checkpoint is seed 1, selected by best validation loss among the 5 final-test seeds.", |
| "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": { |
| "model": "cnn", |
| "channels": [ |
| 32, |
| 64, |
| 128 |
| ], |
| "dropout": 0.1, |
| "parameters": 288618 |
| }, |
| "training": { |
| "optimizer": "adam", |
| "learning_rate": 0.001, |
| "weight_decay": 0.0001, |
| "scheduler": "one_cycle", |
| "label_smoothing": 0.02, |
| "weight_averaging": "ema", |
| "max_epochs": 150, |
| "early_stopping_monitor": "val/loss", |
| "early_stopping_patience": 8, |
| "early_stopping_min_delta": 0.0005 |
| }, |
| "uploaded_checkpoint_metrics": { |
| "seed": 1, |
| "validation_accuracy": 0.996833324432373, |
| "validation_loss": 0.14123967289924622, |
| "test_accuracy": 0.9961000084877014, |
| "test_loss": 0.14772307872772217 |
| }, |
| "five_seed_test_metrics": { |
| "test_accuracy_mean": 0.9961399912834167, |
| "test_accuracy_std": 0.0008018828510757216, |
| "test_loss_mean": 0.14676523208618164, |
| "test_loss_std": 0.002721322756446171, |
| "best_validation_loss_mean": 0.14282282292842866, |
| "best_validation_loss_std": 0.0013830928937782845 |
| } |
| } |
|
|
