Closed marcromeyn closed 1 year ago
This PR builds on the CategoricalOutput and adds ContrastiveOutput. This can be used with dot-product, categorical-prediction or weight-tying.
CategoricalOutput
ContrastiveOutput
The contrastive part of CategoricalOutput doesn't work with torch-script, this is fine since it's during training only.
https://nvidia-merlin.github.io/models/review/pr-1191
github-actions[bot]
commented
1 year ago github-actions[bot]
commented
1 year ago
Goals :soccer:
This PR builds on the
CategoricalOutputand addsContrastiveOutput. This can be used with dot-product, categorical-prediction or weight-tying.Implementation Details :construction:
The contrastive part of
CategoricalOutputdoesn't work with torch-script, this is fine since it's during training only.