tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

How use TensorRT API to add conv1d layers?

Open zhudongwork opened this issue 2 years ago • 2 comments

There is conv1d in my pytorch model. When converting to tensorRT, how to use the tensorRT API to define the conv1d layer?

zhudongwork avatar Sep 15 '22 03:09 zhudongwork

Can you check the tensorrt api doc? I remember the API is called convNd.

wang-xinyu avatar Sep 16 '22 05:09 wang-xinyu

There is conv1d in my pytorch model. When converting to tensorRT, how to use the tensorRT API to define the conv1d layer?

TensorRT貌似不支持1维卷积,可以去英伟达论坛看看,有人提到了并且官方也给出了回答

QIANXUNZDL123 avatar Sep 20 '22 05:09 QIANXUNZDL123

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

stale[bot] avatar Nov 20 '22 07:11 stale[bot]