| { | |
| "best_metric": 0.7789129817933231, | |
| "checkpoint_source": "saved/checkpoints/four_ch_morph/lodo_aptos_convnext_tiny_best.pt", | |
| "class_names": [ | |
| "No DR", | |
| "Mild", | |
| "Moderate", | |
| "Severe", | |
| "Proliferative DR" | |
| ], | |
| "ema_state_dict_present": true, | |
| "epoch": 12, | |
| "global_step": 13742, | |
| "phase_name": "phase2_full_training", | |
| "primary_weights": "lodo_aptos_convnext_tiny_best_ema.safetensors", | |
| "primary_weights_kind": "ema", | |
| "secondary_weights": "lodo_aptos_convnext_tiny_best.safetensors", | |
| "secondary_weights_kind": "raw_model_state_dict", | |
| "training_config": { | |
| "arch": "convnext_tiny", | |
| "cbam_num_stages": 2, | |
| "channels_last": true, | |
| "dropout": 0.4, | |
| "ema_decay": 0.999, | |
| "head_hidden_dim": 512, | |
| "img_size": 384, | |
| "in_chans": 4, | |
| "loss_type": "corn", | |
| "num_thresholds": 4, | |
| "phase_batches": { | |
| "phase1_frozen": 64, | |
| "phase2_full_training": 24 | |
| }, | |
| "use_cbam": true, | |
| "use_class_weighting": true, | |
| "use_mixup": true | |
| } | |
| } |