-
when use torch2trt convert the torch.eq, error occurs.
`mm = torch.eq(mm, 0.)`
mm is tensor and mm.shape = [3136, 1, 3, 3]
File "/media/cfs/torch2trt-master/examples/inpainting/model.py", line 32…
whcjb updated
3 months ago
-
This error occurs when I convert any model, the error log is
`cooperative_groups_helpers.h(87): error: identifier "cudaCGSynchronizeGrid" is undefined`
`1 error detected in the compilation of "…
HHZ94 updated
3 years ago
-
WARNING:root:can't find wrap module for type:, use instead.
[TensorRT] ERROR: Parameter check failed at: ../builder/Layers.cpp::TopKLayer::3499, condition: k > 0 && k 0
../builder/Layers.cpp:3534
…
-
I wanna ask on the use of this library. My data for learning are in a the following shape: `(38079, 300, 8)` and I have a tripleNetwork model. How can I call torch2trt to convert this model using the …
-
Hi,
I'm trying to build engine from pytorch model (resnest200 backboned) with float16 conversion.
Unfortunately I got an error:
`[TensorRT] ERROR: 9: [graphShapeAnalyzer.cpp::throwIfError::1306] …
-
看起来没有定义这个RepeatDimsPluginDynamic plugin, 是需要自己定义么,还是需要从哪里下载插件包。
[TensorRT] ERROR: INVALID_ARGUMENT: getPluginCreator could not find plugin RepeatDimsPluginDynamic version 1
Traceback (most recent …
-
Hello.
SyntaxWarning appeared when I imported torch2trt.
```
C:\DeepLearning\torch2trt-master\torch2trt\dataset.py:61: SyntaxWarning: assertion is always true, perhaps remove parentheses?
asse…
-
I use tons of torch.Tensor.to methods in my model for moving tensor between cpu/cuda, but the method is not supported in torch2trt currently. Will it ever be supported in the future release?
torch.Te…
-
The issue - the `EfficientNet` implementation from the `efficientnet-pytorch` package is not convertible to TensorRT format with torch2trt.
It's very easy to reproduce - am testing this within the …
-
### Description:
I encountered an UnboundLocalError when trying to convert a `MultiLabelSoftMarginLoss` module using torch2trt. The error occurs when the dim parameter is not provided to the torch.me…
Thrsu updated
10 months ago