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

Fix pickling GPU models #632

Merged
merged 3 commits into from Dec 8, 2022
Merged

Fix pickling GPU models #632

merged 3 commits into from Dec 8, 2022

Conversation

benfred
Copy link
Owner

@benfred benfred commented Dec 8, 2022

GPU models weren't default initializing members like _knn appropriately when being loaded from pickle appropriately. Fix and update the unittest to test this out.

GPU models weren't default initializing members like _knn appropriately when
being loaded from pickle appropriately. Fix and update the unittest to test
this out.
@benfred benfred linked an issue Dec 8, 2022 that may be closed by this pull request
@benfred benfred merged commit ec734fb into main Dec 8, 2022
@benfred benfred deleted the pickle_fix branch December 8, 2022 06:39
benfred added a commit that referenced this pull request Dec 8, 2022
GPU models weren't default initializing members like _knn appropriately when
being loaded from pickle appropriately. Fix and update the unittest to test
this out.
@benfred benfred added the bug label Dec 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

ALS missing _knn after unpickling on GPU
1 participant