search

NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
https://nvidia-merlin.github.io/Transformers4Rec/main
Apache License 2.0
1.12k stars 148 forks source link

adding unit test for end-to-end example #669

Closed bbozkaya closed 1 year ago

bbozkaya commented 1 year ago

I am adding a unit test for end-to-end and multi-gpu training example notebook. Currently, multi-gpu notebook is broken. That's why this PR is still work-in-progress.

dependent on: https://github.com/NVIDIA-Merlin/Transformers4Rec/pull/680

review-notebook-app[bot] commented 1 year ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/Transformers4Rec/review/pr-669

rnyak commented 1 year ago

rerun tests