Skip to content

Conversation

benfred
Copy link
Owner

@benfred benfred commented Jun 6, 2023

The GPU ALS model would sometimes return incorrect results with the calculate_training_loss parameter enabled. This happend when the number_of_users * number_of_items was bigger than 2**31 due to an overflow in the loss function calculation.

Fix and add tests that would have caught this bug

Closes #367
Closes #441

The GPU ALS model would sometimes return incorrect results with the
`calculate_training_loss` parameter enabled. This happend when the
number_of_users * number_of_items was bigger than 2**31 due to
an overflow in the loss function calculation.

Fix and tests that would have caught this bug

Closes #367
Closes #441
This was referenced Jun 6, 2023
@benfred benfred merged commit d869ed3 into main Jun 6, 2023
@benfred benfred deleted the fix_gpu_loss branch June 6, 2023 21:08
@benfred benfred added the bug label Jun 13, 2023
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.

different loss from GPU Negative Loss on gpu ALS model
1 participant