Skip to content

fix: CNN3D num_single_blocks off by one; cap channels at 128; version… - #781

Open
anna-grim wants to merge 2 commits into
mainfrom
refactor-vision-backbone
Open

anna-grim wants to merge 2 commits into
mainfrom
refactor-vision-backbone

Conversation

@anna-grim

Copy link
Copy Markdown
Collaborator

…ed configs

Encoder3D used i > num_single_blocks, so the default of 2 gave three single-conv blocks. Now >=. Default max_channels drops from 256 to 128: the 256-channel blocks sat on 8^3 maps and held most of the parameters without adding anything the earlier stages hadn't already encoded.

Both changes alter what a saved config means, so CNN3D configs now carry arch_version (2). CNN3D.upgrade_config maps version-1 configs (no key) onto the current constructor by adding one to num_single_blocks and pinning max_channels=256, so v4a-v4c checkpoints rebuild the exact architecture they trained with. Applied in CNN3D.load and in ArboristVisionMergeDetector.load for the CNN3D backbone.

Anna Grim and others added 2 commits September 21, 2026 15:44
…ed configs

Encoder3D used `i > num_single_blocks`, so the default of 2 gave three
single-conv blocks. Now `>=`. Default max_channels drops from 256 to 128:
the 256-channel blocks sat on 8^3 maps and held most of the parameters
without adding anything the earlier stages hadn't already encoded.

Both changes alter what a saved config means, so CNN3D configs now carry
arch_version (2). CNN3D.upgrade_config maps version-1 configs (no key)
onto the current constructor by adding one to num_single_blocks and
pinning max_channels=256, so v4a-v4c checkpoints rebuild the exact
architecture they trained with. Applied in CNN3D.load and in
ArboristVisionMergeDetector.load for the CNN3D backbone.

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
Vision-only counterpart to ArboristVisionMergeDetector: classifies merge
sites from 3D image patches with no graph input, using the CNN3D or ViT3D
backbone with its built-in output head.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant