- 将pytorch转为onnx的时候,如果有for循环,可以将for循环写死(if torch.onnx.is_in_onnx_export():)
- nonzero: 由于tensorrt没有nonzero这个算子,但是onnx有nonzero这个算子,所以在pytorch中的任何关系索引的算子都会在onnx的时候引入nonzero,所以最好将索引写死
- 遇到’NoneType’ object has no attribute ‘num_bindings’,是因为空间不够了,将condig.max_workspace_size调整为 = 1 << 30
pytorch - onnx - tensorrt
最新推荐文章于 2023-07-16 16:28:47 发布