Skip to content

Dtype parameterization#106

Open
gabrieletijunaityte wants to merge 6 commits into
developfrom
feature/dtypes
Open

Dtype parameterization#106
gabrieletijunaityte wants to merge 6 commits into
developfrom
feature/dtypes

Conversation

@gabrieletijunaityte
Copy link
Copy Markdown
Contributor

@gabrieletijunaityte gabrieletijunaityte commented May 13, 2026 •
edited
Loading

What does this PR do?

  • Added dtype parameter per modality, supported through hydra configs.
  • Added dtype parameter for prediction head and normalisation layer, to unify model weight dtypes.

Before submitting

  • Did you make sure title is self-explanatory and the description concisely explains the PR?
  • Did you make sure your PR does only one thing, instead of bundling different changes together?
  • Did you list all the breaking changes introduced by this pull request?
  • Did you test your PR locally with pytest command?

@gabrieletijunaityte gabrieletijunaityte marked this pull request as draft May 13, 2026 07:53
@gabrieletijunaityte gabrieletijunaityte marked this pull request as ready for review May 13, 2026 08:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@robknapen robknapen Awaiting requested review from robknapen

@vdplasthijs vdplasthijs Awaiting requested review from vdplasthijs

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@gabrieletijunaityte