TensorRT version: 220.127.116.11
CUDA version: 10.0
I want to use use TensorRT with PyTorch, but some layers such as upsamples that are not supported.Can I use custom layers to implement them? If so ,what should I do to implement them?In addition, whether the tensorrt version I am currently using can support this operation?
Looking forward to your reply, thanks .