How to add custom layers for a pytorch network?

AI & Data Science Deep Learning (Training & Inference) TensorRT

J-Penny September 17, 2019, 3:01am 1

Platform：Jetson Xavier
TensorRT version: 5.0.6.3
CUDA version: 10.0

Hello，
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 .

Topic		Replies	Views
Import pytorch model in TensorRT and add custom plugin layer TensorRT	7	5307	May 13, 2019
How to create an custom plugin in TensorRT7 TensorRT tensorrt	1	629	April 10, 2020
Converting Pytorch model through ONNX: "UNSUPPORTED_NODE: No importer registered for op: Greater" TensorRT	2	933	October 12, 2021
Does tensorrt4 support 3D convolution layer and 3D pooling layer of Keras? TensorRT	1	790	December 3, 2018
Tutorial on Onnx model modification / TensorRT plugin development TensorRT	2	291	February 29, 2024
Can I write a Custom Layer Using Python ? TensorRT	1	455	June 6, 2019
Question about converting onnx quantized model to tensorrt TensorRT tensorrt , onnx	2	1090	November 9, 2020
How to avoid trt converting unsupported layers? TensorRT	3	467	May 27, 2020
How to implement BN layers in TensorRT? TensorRT	2	943	November 12, 2018
PyTorch to TensorRT engine with TensorRT 3.0 on Jetson TX2? Jetson TX2	2	2286	October 18, 2021

How to add custom layers for a pytorch network?

Related topics