NVTabular icon indicating copy to clipboard operation
NVTabular copied to clipboard

[FEA] Create PyTorch inference notebook example and add to test_notebooks.py

Open rnyak opened this issue 4 years ago • 0 comments

Is your feature request related to a problem? Please describe. Currently, we do not have a PyT inference example in the test_notebooks.py so we cannot capture pytorch inference errors.

We are getting error issues from users regarding Transformers4Rec PyTorch inference, so it'd be better to catch the errors before users via unit tests, and test notebooks.

Describe the solution you'd like We need to create a PyTorch inference example as we do for movielens with Tensorflow and add it to test_notebooks.py

rnyak avatar Dec 08 '21 16:12 rnyak