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

Use `torch.testing.assert_close` to check model outputs #686

Closed oliverholworthy closed 1 year ago

oliverholworthy commented 1 year ago

Fixes random error in test_sequential_prediction_model_with_ragged_inputs due to slightly different output values.

Use torch.testing.assert_close to check model outputs in test_sequential_prediction_model_with_ragged_inputs

github-actions[bot] commented 1 year ago

Documentation preview

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