{
  "base_checkpoint_path": "/home/rph/scalesurfer/docs/notebooks/01_volume/checkpoints/transunet3d_best.pt",
  "best_val_loss": 0.3957373020654553,
  "description": "Combined FS5/6/7 model; recommended for unknown or mixed versions.",
  "epoch": 10,
  "model": {
    "architectures": [
      "TransUNet3D"
    ],
    "base_volume_shape": [
      256,
      256,
      256
    ],
    "fs_version": 7,
    "library_name": "scalesurfer",
    "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",
    "n_classes": 118,
    "patch_size": [
      16,
      16,
      16
    ],
    "weights_filename": "transunet3d.safetensors"
  },
  "repo_name": "scalesurfer-volume",
  "runtime_fs_version": 7,
  "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_sha256": "c45fbe2a7b30acd6bf6f869216f8e8e57c9802dcc4f211d57abd69af56a492d3",
  "training_fs_versions": [
    5,
    6,
    7
  ],
  "training_objective": "fastsurfer_combined_v254_3d",
  "weights_filename": "transunet3d.safetensors",
  "weights_sha256": "9423cf550f2ae8c74328e2ec2d3fea42123f777204f324bc27ff48a758487ffc"
}