| { |
| "best_val_loss": 0.13997253693722106, |
| "columns_by_group": { |
| "aseg": 119, |
| "global": 64, |
| "lh_aparc": 306, |
| "rh_aparc": 306 |
| }, |
| "config": { |
| "architectures": [ |
| "StatsPredictionModel" |
| ], |
| "encoder": { |
| "architectures": [ |
| "TransUNet3D" |
| ], |
| "base_checkpoint_path": "/home/rph/scalesurfer/docs/notebooks/01_volume/checkpoints/transunet3d_best.pt", |
| "base_volume_shape": [ |
| 256, |
| 256, |
| 256 |
| ], |
| "model_config": { |
| "channels": [ |
| 12, |
| 20, |
| 32, |
| 48, |
| 64, |
| 96 |
| ], |
| "dropout": 0.0, |
| "in_channels": 1, |
| "n_heads": 4, |
| "positional_encoding": "sincos", |
| "task_type": "classification", |
| "transformer_depth": 2 |
| }, |
| "model_type": "scalesurfer-transunet3d-encoder", |
| "n_classes": 118, |
| "patch_size": [ |
| 16, |
| 16, |
| 16 |
| ], |
| "segmentation_loss_kwargs": { |
| "ce_weight": 1.0, |
| "checkpoint_classifier_chunks": true, |
| "dice_weight": 1.0, |
| "edge_weight": 5.0, |
| "include_background": true, |
| "max_class_weight": 5.0, |
| "objective": "fastsurfer_combined_v254_3d" |
| }, |
| "source_checkpoint": "docs/notebooks/01_volume/checkpoints_fastsurfer_v254_staged/combined/combined_fsv567_v254_20260818_102311/transunet3d_best.pt", |
| "source_checkpoint_bytes": 32473907, |
| "source_checkpoint_sha256": "c45fbe2a7b30acd6bf6f869216f8e8e57c9802dcc4f211d57abd69af56a492d3", |
| "training_objective": "fastsurfer_combined_v254_3d" |
| }, |
| "fs_version": null, |
| "library_name": "scalesurfer", |
| "model_type": "scalesurfer-stats-prediction", |
| "model_version": "v567", |
| "source_checkpoint": "data/stats_prediction/checkpoints_stats_canonical_aparc_clean_v5_sum_poolstats_encoder_v567_v254/stats_base_all_20260819_145939/stats_model_best.pt", |
| "stage": "stats_base_all", |
| "stats": { |
| "columns_by_group": { |
| "aseg": 119, |
| "global": 64, |
| "lh_aparc": 306, |
| "rh_aparc": 306 |
| }, |
| "dropout": 0.1, |
| "feature_schema": "region_mean_sum_std_v1", |
| "hidden": 512, |
| "input_dim": 92136, |
| "label_ids": [ |
| 0, |
| 2, |
| 3, |
| 4, |
| 5, |
| 7, |
| 8, |
| 10, |
| 11, |
| 12, |
| 13, |
| 14, |
| 15, |
| 16, |
| 17, |
| 18, |
| 24, |
| 26, |
| 28, |
| 30, |
| 31, |
| 41, |
| 42, |
| 43, |
| 44, |
| 46, |
| 47, |
| 49, |
| 50, |
| 51, |
| 52, |
| 53, |
| 54, |
| 58, |
| 60, |
| 62, |
| 63, |
| 72, |
| 77, |
| 80, |
| 85, |
| 251, |
| 252, |
| 253, |
| 254, |
| 255, |
| 1000, |
| 1001, |
| 1002, |
| 1003, |
| 1004, |
| 1005, |
| 1006, |
| 1007, |
| 1008, |
| 1009, |
| 1010, |
| 1011, |
| 1012, |
| 1013, |
| 1014, |
| 1015, |
| 1016, |
| 1017, |
| 1018, |
| 1019, |
| 1020, |
| 1021, |
| 1022, |
| 1023, |
| 1024, |
| 1025, |
| 1026, |
| 1027, |
| 1028, |
| 1029, |
| 1030, |
| 1031, |
| 1032, |
| 1033, |
| 1034, |
| 1035, |
| 2000, |
| 2001, |
| 2002, |
| 2003, |
| 2004, |
| 2005, |
| 2006, |
| 2007, |
| 2008, |
| 2009, |
| 2010, |
| 2011, |
| 2012, |
| 2013, |
| 2014, |
| 2015, |
| 2016, |
| 2017, |
| 2018, |
| 2019, |
| 2020, |
| 2021, |
| 2022, |
| 2023, |
| 2024, |
| 2025, |
| 2026, |
| 2027, |
| 2028, |
| 2029, |
| 2030, |
| 2031, |
| 2032, |
| 2033, |
| 2034, |
| 2035 |
| ], |
| "label_size_feature_names": [], |
| "pool_features": [ |
| "enc1", |
| "enc2", |
| "enc3", |
| "enc4", |
| "z" |
| ], |
| "pool_stat_names": [ |
| "mean", |
| "sum", |
| "std" |
| ], |
| "target_count": 795 |
| }, |
| "training_fs_versions": [ |
| 5, |
| 6, |
| 7 |
| ], |
| "weights_filename": "stats_model.safetensors" |
| }, |
| "epoch": 195, |
| "feature_schema": "region_mean_sum_std_v1", |
| "fs_version": null, |
| "input_dim": 92136, |
| "model_version": "v567", |
| "repo_name": "scalesurfer-stats", |
| "source_checkpoint": "data/stats_prediction/checkpoints_stats_canonical_aparc_clean_v5_sum_poolstats_encoder_v567_v254/stats_base_all_20260819_145939/stats_model_best.pt", |
| "source_checkpoint_bytes": 770048351, |
| "source_checkpoint_sha256": "65046b187a7c46726f85683cc89377d9ddc94d6abec3c96e1ff976e83f5dba68", |
| "stage": "stats_base_all", |
| "summary_test": [ |
| { |
| "best_epoch": "195", |
| "best_val_loss": "0.13997253693722106", |
| "group": "aseg", |
| "median_abs_percent_error": "6.72153377532959", |
| "n_targets": "119", |
| "n_values": "39699", |
| "normalized_mae": "0.41874403593829795", |
| "split": "test", |
| "stage": "stats_base_all" |
| }, |
| { |
| "best_epoch": "195", |
| "best_val_loss": "0.13997253693722106", |
| "group": "global", |
| "median_abs_percent_error": "1.0538854598999023", |
| "n_targets": "64", |
| "n_values": "16070", |
| "normalized_mae": "0.15233277114068536", |
| "split": "test", |
| "stage": "stats_base_all" |
| }, |
| { |
| "best_epoch": "195", |
| "best_val_loss": "0.13997253693722106", |
| "group": "lh_aparc", |
| "median_abs_percent_error": "7.2162065505981445", |
| "n_targets": "306", |
| "n_values": "106742", |
| "normalized_mae": "0.4084990393382036", |
| "split": "test", |
| "stage": "stats_base_all" |
| }, |
| { |
| "best_epoch": "195", |
| "best_val_loss": "0.13997253693722106", |
| "group": "rh_aparc", |
| "median_abs_percent_error": "7.313578844070435", |
| "n_targets": "306", |
| "n_values": "106738", |
| "normalized_mae": "0.4122215262526054", |
| "split": "test", |
| "stage": "stats_base_all" |
| } |
| ], |
| "target_count": 795, |
| "training_fs_versions": [ |
| 5, |
| 6, |
| 7 |
| ], |
| "weights_bytes": 770050564, |
| "weights_filename": "stats_model.safetensors", |
| "weights_sha256": "69713bf0d5ae87ab5da5268861b8daaa2d103d52d86f975ac384f054746d0c60" |
| } |
|
|
