Skip to content

Refactor encoder output#739

Open
anna-grim wants to merge 6 commits into
mainfrom
refactor-encoder-output
Open

Refactor encoder output#739
anna-grim wants to merge 6 commits into
mainfrom
refactor-encoder-output

Conversation

@anna-grim

Copy link
Copy Markdown
Collaborator

No description provided.

anna-grim added 6 commits July 21, 2026 22:46
- Fix merge site index-0 label bug (idx > 0 → idx >= 0) in get_site
- Fix __len__ to compute count analytically instead of re-sampling each call
- Cache fragment cable-length traversal in _compute_fragment_stats so
  find_fragments_to_search and estimate_iterations share one graph walk
- Replace per-node is_near_leaf BFS with single multi-source BFS
  (compute_near_leaf_nodes) per component, reducing O(N²) to O(N)
- Remove SEResConvBlock3D, SqueezeExcite3D, ConvNeXtDownBlock3D,
  ConvNeXtBlock3D, and CenterWeightedPool3D — underperforming on this task
- Replace center-weighted pooling with standard avg+max pool at last 3 stages
- Fix checkpoint save format to include config alongside state_dict so
  NewCNN3D.load can reconstruct the model without manual architecture setup
- Fix CosineAnnealingLR T_max=25 → T_max=max_epochs for single decay
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