Skip to content

Fix hardcoded in_channels in UnetOutBlock to use feat_size[0]#6

Open
skaliy wants to merge 1 commit into
ge-xing:mainfrom
skaliy:fix/dynamic-out-block-channels
Open

Fix hardcoded in_channels in UnetOutBlock to use feat_size[0]#6
skaliy wants to merge 1 commit into
ge-xing:mainfrom
skaliy:fix/dynamic-out-block-channels

Conversation

@skaliy
Copy link
Copy Markdown

@skaliy skaliy commented Mar 2, 2026

Replace hardcoded value of 48 with self.feat_size[0] so the output block works correctly when SegMamba is instantiated with custom feat_size values.

Replace hardcoded value of 48 with self.feat_size[0] so the output block
works correctly when SegMamba is instantiated with custom feat_size values.

Co-Authored-By: Claude Opus 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