{
  "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"
}