Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Changing nn ToDtype to use generic on method #719

Merged
merged 1 commit into from
Apr 20, 2023
Merged

Conversation

coreylowman
Copy link
Owner

Related to #423

This is useful for unit tests when f16 is added, so we can do model.to_dtype::<TestDtype>(). It also makes it more usable.

@coreylowman coreylowman merged commit a48743b into main Apr 20, 2023
4 checks passed
@coreylowman coreylowman deleted the nn-to-dtype branch April 20, 2023 15:19
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.

None yet

1 participant