NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Apache License 2.0
758
stars
113
forks
source link
Quick-start for Ranking: Fix bug when saving model trained with single task and moved util classes to Models #967
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.
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.