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.07k stars 142 forks source link

Fix transformer and error on example when CI uses single-GPU #757

Closed gabrielspmoreira closed 9 months ago

gabrielspmoreira commented 9 months ago

This PR fixes CI notebook tests to work with single-GPU CI server. It includes the commits from PR #755 (fix on going errors in transformers unit tests) by @jperez999

copy-pr-bot[bot] commented 9 months ago

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

review-notebook-app[bot] commented 9 months ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

github-actions[bot] commented 9 months ago

Documentation preview

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

gabrielspmoreira commented 9 months ago

/ok to test