| { |
| "model": { |
| "width": 32, |
| "blocks": [ |
| 2, |
| 2, |
| 2, |
| 2 |
| ], |
| "num_color": 7, |
| "num_type": 7, |
| "dropout": 0.2 |
| }, |
| "labels": { |
| "color": [ |
| "White", |
| "Blue", |
| "Black", |
| "Red", |
| "Green", |
| "Colorless", |
| "Multicolor" |
| ], |
| "type": [ |
| "Creature", |
| "Planeswalker", |
| "Land", |
| "Artifact", |
| "Enchantment", |
| "Instant", |
| "Sorcery" |
| ] |
| }, |
| "normalization": { |
| "mean": [ |
| 0.412272, |
| 0.37507, |
| 0.34918 |
| ], |
| "std": [ |
| 0.251613, |
| 0.230892, |
| 0.231858 |
| ] |
| }, |
| "image_size": [ |
| 160, |
| 224 |
| ], |
| "dials": { |
| "run_name": "light-aug", |
| "epochs": 30, |
| "batch_size": 256, |
| "lr": 0.002, |
| "weight_decay": 0.05, |
| "warmup_frac": 0.05, |
| "width": 32, |
| "blocks": [ |
| 2, |
| 2, |
| 2, |
| 2 |
| ], |
| "dropout": 0.2, |
| "loss": "ce", |
| "label_smoothing": 0.1, |
| "focal_gamma": 2.0, |
| "type_loss_weight": 1.0, |
| "color_loss_weight": 1.0, |
| "img_size": [ |
| 160, |
| 224 |
| ], |
| "eval_every_epochs": 1, |
| "patience": 6, |
| "grad_clip": 1.0, |
| "hflip": false, |
| "aug": "light", |
| "limit": null, |
| "seed": 42 |
| }, |
| "params": 2800366, |
| "dtype": "bfloat16", |
| "train_state": { |
| "epoch": 20, |
| "best_epoch": 20, |
| "best_score": 0.3503234434514614, |
| "best_val": { |
| "val_loss": 2.9443500779889558, |
| "val_color_acc": 0.3781546537452049, |
| "val_color_macro_f1": 0.368955477464954, |
| "val_type_acc": 0.5659196446598022, |
| "val_type_macro_f1": 0.3316914094379687, |
| "val_score": 0.3503234434514614 |
| } |
| }, |
| "source_checkpoint": "/home/nathank/BITS/dev/MTG Card Classifier/checkpoints/light-aug/best", |
| "exported_at": "2026-10-23 20:04:46", |
| "metrics_test": { |
| "split": "test", |
| "n": 3401, |
| "evaluated_at": "2026-10-23 20:04:39", |
| "color": { |
| "accuracy": 0.3684210526315789, |
| "top2_accuracy": 0.5698324022346368, |
| "macro_f1": 0.3450303818057278, |
| "ece": 0.04191294569227494 |
| }, |
| "type": { |
| "accuracy": 0.5292561011467215, |
| "top2_accuracy": 0.6745074977947663, |
| "macro_f1": 0.2856992909416024, |
| "ece": 0.05380913312909744 |
| } |
| }, |
| "latency": { |
| "samples": 100, |
| "gpu_bf16": { |
| "n": 100, |
| "mean_s": 0.00373, |
| "p50_s": 0.00368, |
| "p95_s": 0.00424, |
| "max_s": 0.00475, |
| "ratio_to_benchmark": 0.05, |
| "device": "NVIDIA GeForce RTX 3090 Ti" |
| }, |
| "cpu_fp32": { |
| "n": 100, |
| "mean_s": 0.0052, |
| "p50_s": 0.00505, |
| "p95_s": 0.00601, |
| "max_s": 0.00809, |
| "ratio_to_benchmark": 0.07, |
| "threads": 16 |
| }, |
| "benchmark_s": 0.074 |
| }, |
| "model_py_sha1": "8d93607d00c223dba206254a9a03c1e24f23bd97" |
| } |
|
|
