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.11k stars 146 forks source link

Fixing the projection layer when using weight tying and dim from Transformer output and item embedding differs #689

Closed gabrielspmoreira closed 1 year ago

gabrielspmoreira commented 1 year ago

Fixes the projection when using weight tying, as it should not use an activation function, as found by @bschifferer

github-actions[bot] commented 1 year ago

Documentation preview

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