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

[QST] Help with creating two tower model with transformers. #763

Open vivpra89 opened 7 months ago

vivpra89 commented 7 months ago

❓ Questions & Help

Details

I have two separate sequences and want to create two towers and then concat the output layers of two transformer towers. Can you please help me with this implementation? Is it possible to do it with transformer4rec? Im am not sure how to contact the output layers of the two towers @gabrielspmoreira @rnyak @oliverholworthy @benfred

karunaahuja commented 4 months ago

@vivpra89 - could you figure this out?