Yang0822's picture
Release verified SlimFormer++ acdc checkpoint and model card
304bb5d verified
Raw History Blame Contribute Delete
2.34 kB
{
"alias": "acdc",
"dataset": "Dataset003_ACDC",
"dataset_id": 3,
"model_folder": "nnUNetTrainer_SlimFormerPlusPlus__nnUNetPlans__3d_fullres",
"release_tag": "acdc-best-v1",
"checkpoint": "checkpoint_best.pth",
"checkpoint_sha256": "64f57fd529fdb3518fd9a0dc3452b584ed8bcf73b322702bdb1b9f913d350256",
"checkpoint_current_epoch": 254,
"trainer_sha256": "ecb4ff6bc99d724248e8215a0c435e8e30cca4e96a122588317ed11a13bb1245",
"split": {
"fold": 0,
"train_cases": 240,
"validation_cases": 60,
"evaluation_set": "single-fold validation; not an independent test set",
"train_patients": 120,
"validation_patients": 30,
"patient_overlap": 0
},
"training": {
"run": "slimformer-plusplus-ep1200-b8",
"completed_epochs": 1200,
"optimizer_steps_per_epoch": 250,
"validation_iterations_per_epoch": 50,
"global_batch": 8,
"gpus": 4,
"per_gpu_batch": 2,
"initialization_seed": 20260921,
"optimizer": "SGD",
"initial_lr": 0.01,
"momentum": 0.99,
"nesterov": true,
"weight_decay": 3e-05,
"no_weight_decay": [
"theta",
"relation_bias"
],
"scheduler": "PolyLR",
"poly_exponent": 0.9,
"deep_supervision": false,
"compile": false,
"best_selection": "highest validation EMA pseudo-Dice"
},
"inference": {
"configuration": "3d_fullres",
"patch": [
20,
256,
224
],
"tile_step_size": 0.5,
"gaussian_weighting": true,
"mirroring_axes": [
0,
1,
2
],
"ensemble": false,
"postprocessing": false
},
"metrics": {
"aggregation": "mean over cases per foreground label, then mean over labels",
"Dice_IoU_units": "fraction",
"HD95_ASD_units": "mm",
"implementation": "reproduce.py; project evaluation convention"
},
"provenance": {
"network_weights_verified_equal_to_release": true,
"plans_and_dataset_verified_equal_to_release": true,
"source_summary_sha256": "51bc15b46358084a528a26801e35cb0770f74f3d79a19921487d8f41fa544cff",
"source_surface_summary_sha256": "54a342bb933d601e3843759c6c40bb0107fc9b2635fae48543d1d0222c855738",
"source_splits_sha256": "d8df54d3b50972dab7a033e589359c917b0ccc17ad44e2def5fe8a49a7e837e0",
"efficiency_source": "measure_slimformer_plusplus_mixswap_ft_cost.py, baseline arm"
}
}