Repository: NVIDIA-Merlin/NVTabular

Issue:
[BUG] getting error from export_pytorch_ensemble()
**Describe the bug**
Getting the following error while running `export_pytorch_ensemble()` in cell 26 in this [notebook](https://github.com/NVIDIA-Merlin/Transformers4Rec/blob/main/examples/end-to-end-session-based/end-to-end-session-based-with-Yoochoose.ipynb).

```
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_783/617147693.py in <module>
      1 from nvtabular.inference.triton import export_pytorch_ensemble
----> 2 export_pytorch_ensemble(
      3     model,
      4     workflow,
      5     sparse_max=recsys_trainer.get_train_dataloader().dataset.sparse_max,

/nvtabular/nvtabular/inference/triton/ensemble.py in export_pytorch_ensemble(model, workflow, sparse_max, name, model_path, label_columns, use_fix_dtypes, version, nvtabular_backend)
    139     # generate the TF saved model
    140     pt_path = os.path.join(model_path, name + "_pt")
--> 141     pt_config = export_pytorch_model(
    142         model, workflow, sparse_max, name + "_pt", pt_path, use_fix_dtypes, version=version
    143     )

/nvtabular/nvtabular/inference/triton/ensemble.py in export_pytorch_model(model, workflow, sparse_max, name, output_path, use_fix_dtypes, version, backend)
    570     config.output.append(
    571         model_config.ModelOutput(
--> 572             name="output", data_type=_convert_pytorch_dtype(dtype), dims=[-1, dims]
    573         )
    574     )

/nvtabular/nvtabular/inference/triton/ensemble.py in _convert_pytorch_dtype(dtype)
    787         torch.int16: model_config.TYPE_INT16,
    788         torch.int8: model_config.TYPE_INT8,
--> 789         torch.uint64: model_config.TYPE_UINT64,
    790         torch.uint32: model_config.TYPE_UINT32,
    791         torch.uint16: model_config.TYPE_UINT16,

AttributeError: module 'torch' has no attribute 'uint64'
```

Looks like we need to update `_convert_pytorch_dtype(dtype)` function.  In [this doc](https://pytorch.org/docs/stable/tensors.html) `uint64` does not exist: 

Container: `merlin-pytorch-training:21.11` pulled the latest main branch
Pytorch: `1.10.10` public pytorch
TF4rec: v0.1.2


Discussion:

Resolving patch:
diff --git a/nvtabular/inference/triton/ensemble.py b/nvtabular/inference/triton/ensemble.py
--- a/nvtabular/inference/triton/ensemble.py
+++ b/nvtabular/inference/triton/ensemble.py
@@ -786,9 +786,6 @@ def _convert_pytorch_dtype(dtype):
         torch.int32: model_config.TYPE_INT32,
         torch.int16: model_config.TYPE_INT16,
         torch.int8: model_config.TYPE_INT8,
-        torch.uint64: model_config.TYPE_UINT64,
-        torch.uint32: model_config.TYPE_UINT32,
-        torch.uint16: model_config.TYPE_UINT16,
         torch.uint8: model_config.TYPE_UINT8,
         torch.bool: model_config.TYPE_BOOL,
     }
@@ -819,9 +816,6 @@ def _convert_string2pytorch_dtype(dtype):
         "TYPE_INT32": torch.int32,
         "TYPE_INT16": torch.int16,
         "TYPE_INT8": torch.int8,
-        "TYPE_UINT64": torch.uint64,
-        "TYPE_UINT32": torch.uint32,
-        "TYPE_UINT16": torch.uint16,
         "TYPE_UINT8": torch.uint8,
         "TYPE_BOOL": torch.bool,
     }


Test patch: