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