search

NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

Match shape and dtype in loss calculation in pytorch backend #1156

Closed edknv closed 1 year ago

edknv commented 1 year ago

This PR uncomments a basic use case of training a model with TabularInputBlock using the Lightning trainer.

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1156

edknv commented 1 year ago

Redundant after #1171.