Skip to content

A few changes related to UNETR ViT backbones - #190

Merged
constantinpape merged 5 commits into
mainfrom
update-backbones
Oct 29, 2026
Merged

constantinpape merged 5 commits into
mainfrom
update-backbones

Conversation

@stmartineau99

@stmartineau99 stmartineau99 commented Oct 29, 2026 •

Copy link
Copy Markdown
Contributor

Changes

  • dinov3 backbone: _get_checkpoint loads the encoder via microsam2 get_vfm_model, like dinov2
  • this means that training with ViT backbones is dependent on having microsam's dev branch installed for now
  • bf16 inference: new dtype arg in get_prediction, torch.bfloat16 runs the model under autocast

Tests

  • added test_prediction_bfloat16 test to compare bfloat16 autocast predictions with default float32

@constantinpape
constantinpape merged commit ffad121 into main Oct 29, 2026
3 checks passed
@constantinpape
constantinpape deleted the update-backbones branch October 29, 2026 20:27
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.

2 participants