Open
Description
Thank you for your great work. I am trying to follow your instruction to convert UNET to tensorrt format, however, I got the following errors:
- when converting the model to onnx: I got stuck because the att:ScaledDotProductAttention is not supported. I found the solution on the internet by adding a customized function for scaleddotproductattention. However, I once again encounter this issue when trying to convert the model from onnx to trt. I just want to ask if you have ever encountered this error. And if yes, how did you manage to fix it? Thank you!
Metadata
Metadata
Assignees
Labels
No labels