jskvrna commited on
Commit
ecb8b90
·
1 Parent(s): a3f2b75

feat: Update ensemble_merge_m default value to 0.5 and adjust help description for consistency

Browse files
script.py CHANGED
@@ -334,7 +334,7 @@ class Stage2Config:
334
  density_planarity_min_points: int = 12,
335
  density_min_per_voxel: int = 0,
336
  ensemble_n: int = 1,
337
- ensemble_merge_m: float = 0.4,
338
  ensemble_mode: str = "medoid",
339
  ensemble_strategy: str = "union_hull",
340
  ensemble_refine_positions: bool = True,
@@ -1803,12 +1803,11 @@ def parse_args(argv=None):
1803
  "--ensemble_strategy union_hull this is ~1.2x wall "
1804
  "since stage-2 shares one encoder pass at B=N). "
1805
  "Set to 1 to disable.")
1806
- p.add_argument("--ensemble_merge_m", type=float, default=0.4,
1807
  help="Vertex match threshold in metres for ensemble "
1808
  "selection (medoid Hungarian distance, position "
1809
  "refinement correspondents, consensus clustering). "
1810
- "Default 0.4. The HSS metric uses 0.5; pass 0.5 "
1811
- "to align with it.")
1812
  p.add_argument("--ensemble_strategy",
1813
  choices=["union_hull", "per_seed"], default="union_hull",
1814
  help="How stage-2 is run across the N ensemble members. "
@@ -1820,12 +1819,12 @@ def parse_args(argv=None):
1820
  "robust if union-hull occasionally fails.")
1821
  p.add_argument("--ensemble_mode",
1822
  choices=["medoid", "consensus", "confidence"],
1823
- default="confidence",
1824
  help="How to combine the N ensemble runs. "
1825
- "'confidence' (default) picks the run with the "
1826
- "highest mean validity logit. 'medoid' picks the "
1827
- "run most similar to the others by Hungarian-"
1828
- "matched distance. 'consensus' merges vertices and "
1829
  "majority-votes edges (lost recall in earlier tests).")
1830
  p.add_argument("--no_position_refine", action="store_true",
1831
  help="Disable position refinement. By default, after the "
 
334
  density_planarity_min_points: int = 12,
335
  density_min_per_voxel: int = 0,
336
  ensemble_n: int = 1,
337
+ ensemble_merge_m: float = 0.5,
338
  ensemble_mode: str = "medoid",
339
  ensemble_strategy: str = "union_hull",
340
  ensemble_refine_positions: bool = True,
 
1803
  "--ensemble_strategy union_hull this is ~1.2x wall "
1804
  "since stage-2 shares one encoder pass at B=N). "
1805
  "Set to 1 to disable.")
1806
+ p.add_argument("--ensemble_merge_m", type=float, default=0.5,
1807
  help="Vertex match threshold in metres for ensemble "
1808
  "selection (medoid Hungarian distance, position "
1809
  "refinement correspondents, consensus clustering). "
1810
+ "Default 0.5 matches the HSS metric's vert_thresh.")
 
1811
  p.add_argument("--ensemble_strategy",
1812
  choices=["union_hull", "per_seed"], default="union_hull",
1813
  help="How stage-2 is run across the N ensemble members. "
 
1819
  "robust if union-hull occasionally fails.")
1820
  p.add_argument("--ensemble_mode",
1821
  choices=["medoid", "consensus", "confidence"],
1822
+ default="medoid",
1823
  help="How to combine the N ensemble runs. "
1824
+ "'medoid' (default) picks the run most similar to "
1825
+ "the others by Hungarian-matched distance. "
1826
+ "'confidence' picks the run with the highest mean "
1827
+ "validity logit. 'consensus' merges vertices and "
1828
  "majority-votes edges (lost recall in earlier tests).")
1829
  p.add_argument("--no_position_refine", action="store_true",
1830
  help="Disable position refinement. By default, after the "
test_checkpoint_stage2.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3918551171c82f7ac65f24511ff3c58af5c1e272b8838b1a5acf2c4b4595d5e4
3
- size 408301733
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9ef48a502e3b9e4324180362f996ff3927e4bc55ae5ee8e8227543400d3ab1b
3
+ size 408333399
test_checkpoint_stage2_best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3918551171c82f7ac65f24511ff3c58af5c1e272b8838b1a5acf2c4b4595d5e4
3
+ size 408301733
test_checkpoint_stage2_new.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a4dc400fdc78704e57aa3360e76824ff0f0fbf66a1ef3f0bf07673df0173a6b
3
+ size 408301925