Skip to content

Fix DPT decoder bugs#26

Open
bingyic wants to merge 1 commit intoadd-decoders-modulefrom
fix-dpt-decoders
Open

Fix DPT decoder bugs#26
bingyic wants to merge 1 commit intoadd-decoders-modulefrom
fix-dpt-decoders

Conversation

@bingyic
Copy link
Copy Markdown
Collaborator

@bingyic bingyic commented Apr 22, 2026

This PR applies fixes to the DPT decoders introduced in PR #24.

Changes:

  • Add F.relu() after DPTHead project conv to match Scenic's output_activation=True default.
  • Fix DepthDecoder to route through parent's nn.Linear head instead of bypassing it.
  • Register bin_centers as a buffer with configurable num_depth_bins.
  • Add weight key remapping for all decoder types.

- Add F.relu() after DPTHead project conv to match Scenic's output_activation=True default
- Fix DepthDecoder to route through parent's nn.Linear head instead of bypassing it
- Register bin_centers as a buffer with configurable num_depth_bins
- Add weight key remapping for all decoder types
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