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

Quick-start for Ranking: Fix bug when saving model trained with single task and moved util classes to Models #967

Merged
merged 2 commits into from
May 10, 2023

Conversation

gabrielspmoreira
Copy link
Member

This PR fixes a bug that was occurring when trying to save a model trained for single-task learning.
It also extracts the following util classes from the Quick-Start scrips, which are added in #1085 to Merlin Models:

  • LogLossMetric
  • WandbLogger
  • ExamplesPerSecondCallback

Important: This PR can only be merged after #1085 is merged.

@gabrielspmoreira gabrielspmoreira requested a review from rnyak May 8, 2023 20:19
@gabrielspmoreira gabrielspmoreira self-assigned this May 8, 2023
@gabrielspmoreira gabrielspmoreira added bug Something isn't working enhancement New feature or request labels May 8, 2023
@gabrielspmoreira gabrielspmoreira added this to the Merlin 23.05 milestone May 8, 2023
@github-actions
Copy link

github-actions bot commented May 8, 2023

Documentation preview

https://nvidia-merlin.github.io/Merlin/review/pr-967

@gabrielspmoreira gabrielspmoreira force-pushed the tf/qq_extract_logloss_callbacks branch from ef15aba to a64b9d6 Compare May 8, 2023 20:25
@rnyak rnyak merged commit 4cd7715 into main May 10, 2023
@rnyak rnyak deleted the tf/qq_extract_logloss_callbacks branch May 10, 2023 21:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants