Upload ScaleSurfer volume model

README.md

@@ -23,7 +23,7 @@ This repository contains the ScaleSurfer 3D TransUNet volume segmentation checkp
 
 | Split counts | Best validation CE | Final test CE |
 | --- | ---: | ---: |
-| train=1592, val=202, test=202 | 0.036986797236570036 | 0.03448366988819129 |
+| train=1592, val=202, test=202 | 0.021963781686407505 | 0.020560515551173157 |
 
 ## Loading
 

config.json

@@ -3,9 +3,9 @@
     "TransUNet3D"
   ],
   "base_volume_shape": [
-    208,
-    240,
-    192
+    256,
+    256,
+    256
   ],
   "fs_version": 6,
   "library_name": "scalesurfer",
@@ -32,6 +32,6 @@
     16,
     16
   ],
-  "source_checkpoint": "docs/notebooks/checkpoints_fsv6/fsv6_20260402_021927/transunet3d_best.pt",
+  "source_checkpoint": "docs/notebooks/01_volume/checkpoints_fsv6/fsv6_20260421_050137/transunet3d_best.pt",
   "weights_filename": "transunet3d.safetensors"
 }

history.csv

@@ -1,6 +1,26 @@
 epoch,train_ce,val_ce,lr,micro_bs,patch_chunk,window,sec,alloc_gb,peak_gb
-1,0.035362626701630866,0.03875425550164563,0.0004849699476419097,1,192,,643.185310125351,0.05597686767578125,14.427436351776123
-2,0.03527034926248713,0.03800764117426801,0.0003752816052402873,1,192,,578.3788032531738,0.05597686767578125,14.427436351776123
-3,0.03426110046481951,0.03745010879702202,0.00020719874099207078,1,192,,647.2710845470428,0.05597686767578125,14.427436351776123
-4,0.033469451206199485,0.037100644671533366,5.937973142222941e-05,1,192,,578.8744940757751,0.05597686767578125,14.427436351776123
-5,0.03300896497346147,0.036986797236570036,1e-06,1,192,,559.2231624126434,0.05597686767578125,14.427436351776123
+1,0.02335054350233655,0.02417138475223933,0.00020070035193564607,1,192,,964.5021080970764,0.05683135986328125,25.069576263427734
+2,0.023414618578592537,0.02394541243257204,0.00040027526395173456,1,192,,961.7268743515015,0.05683135986328125,25.069576263427734
+3,0.023414214441961345,0.02399261479293651,0.0004993929770491203,1,192,,960.9014513492584,0.05683135986328125,25.069576263427734
+4,0.023178281748958873,0.02383921726486913,0.0004945499506918455,1,192,,961.8910174369812,0.05683135986328125,25.069576263427734
+5,0.023018396423812336,0.023739407147917124,0.0004849566371647871,1,192,,961.2872877120972,0.05683135986328125,25.069576263427734
+6,0.022733044000442,0.023461702279746532,0.00047079976461123787,1,192,,961.283205986023,0.05683135986328125,25.069576263427734
+7,0.02254329786456227,0.023374260970308345,0.0004523548881372924,1,192,,961.2581424713135,0.05683135986328125,25.069576263427734
+8,0.022359090486669002,0.023120368784754583,0.0004299810262955632,1,192,,960.1396019458771,0.05683135986328125,25.069576263427734
+9,0.02216853090520365,0.022988549267670307,0.00040411367300253705,1,192,,960.5774521827698,0.05683135986328125,25.069576263427734
+10,0.021957640613398225,0.023196710988252173,0.00037525632090845137,1,192,,1087.1651611328125,0.05683135986328125,25.069576263427734
+11,0.0218391964413427,0.02274220524411095,0.0003439706612122505,1,192,,963.4231781959534,0.05683135986328125,25.069576263427734
+12,0.021672325122148492,0.022801098570522694,0.00031086565067639987,1,192,,963.7764203548431,0.05683135986328125,25.069576263427734
+13,0.02150804176868076,0.02275178881620269,0.00027658565864561004,1,192,,963.7171375751495,0.05683135986328125,25.069576263427734
+14,0.021380475515617954,0.022596857307644765,0.00024179792478070669,1,192,,963.267707824707,0.05683135986328125,25.069576263427734
+15,0.02120006601323241,0.022470528577998427,0.00020717957163539588,1,192,,962.9111084938049,0.05683135986328125,25.069576263427734
+16,0.021074810005036037,0.022551251026700334,0.00017340442486839616,1,192,,963.4325630664825,0.05683135986328125,25.069576263427734
+17,0.020977196517786203,0.022390032809522776,0.00014112989762767204,1,192,,964.7174162864685,0.05683135986328125,25.069576263427734
+18,0.020853097955646528,0.0222126120553896,0.00011098419439442805,1,192,,966.3648147583008,0.05683135986328125,25.069576263427734
+19,0.020752861933685526,0.022147828625459776,8.35540833564351e-05,1,192,,967.8573350906372,0.05683135986328125,25.069576263427734
+20,0.020668065870247457,0.022084409039061847,5.9373475314166866e-05,1,192,,963.6563680171967,0.05683135986328125,25.069576263427734
+21,0.020592384568358803,0.02203641964657472,3.891303142455132e-05,1,192,,967.312037229538,0.05683135986328125,25.069576263427734
+22,0.020539809889988395,0.022002285825881627,2.2571002061421042e-05,1,192,,964.2529513835907,0.05683135986328125,25.069576263427734
+23,0.020494568028202192,0.021973652520797925,1.0665475108800885e-05,1,192,,967.1072909832001,0.05683135986328125,25.069576263427734
+24,0.02046698507736364,0.02196531736817543,3.428184569390664e-06,1,192,,966.93497133255,0.05683135986328125,25.069576263427734
+25,0.020450942186108368,0.021963781686407505,1e-06,1,192,,967.2839677333832,0.05683135986328125,25.069576263427734

metadata.json

@@ -1,5 +1,5 @@
 {
-  "best_val": 0.036986797236570036,
+  "best_val": 0.021963781686407505,
   "data_summary": {
     "n_test_x_files": 202,
     "n_test_y_files": 202,
@@ -8,16 +8,16 @@
     "n_val_x_files": 202,
     "n_val_y_files": 202
   },
-  "epoch": 5,
+  "epoch": 25,
   "fs_version": 6,
   "model": {
     "architectures": [
       "TransUNet3D"
     ],
     "base_volume_shape": [
-      208,
-      240,
-      192
+      256,
+      256,
+      256
     ],
     "fs_version": 6,
     "library_name": "scalesurfer",
@@ -44,29 +44,29 @@
       16,
       16
     ],
-    "source_checkpoint": "docs/notebooks/checkpoints_fsv6/fsv6_20260402_021927/transunet3d_best.pt",
+    "source_checkpoint": "docs/notebooks/01_volume/checkpoints_fsv6/fsv6_20260421_050137/transunet3d_best.pt",
     "weights_filename": "transunet3d.safetensors"
   },
   "repo_name": "scalesurfer-v6",
   "runtime_micro_bs": 1,
   "runtime_patch_chunk": 192,
   "runtime_window": null,
-  "source_checkpoint": "docs/notebooks/checkpoints_fsv6/fsv6_20260402_021927/transunet3d_best.pt",
-  "source_checkpoint_bytes": 32054973,
-  "source_checkpoint_sha256": "e1c51c4d00a8afa50440cd2b2bb6ce7735634f1cdf31316a2188f8acf02ea3b2",
+  "source_checkpoint": "docs/notebooks/01_volume/checkpoints_fsv6/fsv6_20260421_050137/transunet3d_best.pt",
+  "source_checkpoint_bytes": 32001405,
+  "source_checkpoint_sha256": "42f86fbd177fd6817d7e2c05a15b2b882c503380df50c47c840c9b31755c6a39",
   "training_summary": {
-    "best_epoch": "5",
-    "best_val_ce": "0.036986797236570036",
-    "cosine_total_steps": "7955",
-    "final_test_ce": "0.03448366988819129",
+    "best_epoch": "25",
+    "best_val_ce": "0.021963781686407505",
+    "cosine_total_steps": "39775",
+    "final_test_ce": "0.020560515551173157",
     "fs_major": "6",
     "n_test": "202",
     "n_train": "1592",
     "n_val": "202",
     "run_name": "fsv6",
-    "warmup_steps": "796"
+    "warmup_steps": "3978"
   },
-  "weights_bytes": 10493200,
+  "weights_bytes": 10493216,
   "weights_filename": "transunet3d.safetensors",
-  "weights_sha256": "fd3fa7bd3a7aa3460c0feeafc1f147a9b0d70877969a45f6c2b1cc1175f48975"
+  "weights_sha256": "e3fee0400256bf81dce1358bc74cd10f40b5932646d70625805dc83c4a20fb8a"
 }

transunet3d.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fd3fa7bd3a7aa3460c0feeafc1f147a9b0d70877969a45f6c2b1cc1175f48975
-size 10493200
+oid sha256:e3fee0400256bf81dce1358bc74cd10f40b5932646d70625805dc83c4a20fb8a
+size 10493216