| { |
| "dataset": { |
| "raw_dir": "/content/drive/MyDrive/nsl_raw", |
| "num_raw_images": 4815, |
| "num_valid_labeled_images": 4812, |
| "num_classes": 137, |
| "min_total_per_class": 1, |
| "dropped_classes": [], |
| "split_protocol": "official source test split; validation split from source train by class", |
| "split_counts": { |
| "train": 3581, |
| "val": 636, |
| "test": 595 |
| }, |
| "classes_per_split": { |
| "test": 136, |
| "train": 137, |
| "val": 136 |
| }, |
| "duplicate_hashes_crossing_splits": 0 |
| }, |
| "vit_classification": { |
| "model": "vit_base_patch16_224", |
| "input_mode": "crop", |
| "epochs_requested": 40, |
| "head_warmup_epochs": 3, |
| "best_checkpoint": "/content/nsl_vit_fixed_work/runs/vit_b16_nsl_fixed_best.pt", |
| "params": 85904009, |
| "val": { |
| "loss": 0.8947618052644549, |
| "top1": 0.9261006289308176, |
| "top5": 0.9638364779874213, |
| "macro_f1": 0.9024900968879636, |
| "weighted_f1": 0.9203976265041456 |
| }, |
| "test": { |
| "loss": 0.8799869286913832, |
| "top1": 0.9210084033613445, |
| "top5": 0.9697478991596639, |
| "macro_f1": 0.9066332142953599, |
| "weighted_f1": 0.9183226019280669 |
| } |
| } |
| } |
