feat: Update ensemble_merge_m default value to 0.5 and adjust help description for consistency
Browse files- script.py +8 -9
- test_checkpoint_stage2.pth +2 -2
- test_checkpoint_stage2_best.pth +3 -0
- test_checkpoint_stage2_new.pth +3 -0
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.
|
| 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.
|
| 1807 |
help="Vertex match threshold in metres for ensemble "
|
| 1808 |
"selection (medoid Hungarian distance, position "
|
| 1809 |
"refinement correspondents, consensus clustering). "
|
| 1810 |
-
"Default 0.
|
| 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="
|
| 1824 |
help="How to combine the N ensemble runs. "
|
| 1825 |
-
"'
|
| 1826 |
-
"
|
| 1827 |
-
"
|
| 1828 |
-
"
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 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
|