Hi all,
I am trying to export a pre-trained model using Torchscript. The example in HuggingFace docs for this task has a list of tensors as input for forward
. PyTorch document on how to start with TorchScript have an example with a single Tensor.
How do I proceed with providing input to the model with a list of tensors as input?
HuggingFace reference: https://huggingface.co/transformers/torchscript.html
PyTorch reference: https://pytorch.org/tutorials/advanced/cpp_export.html