UpsamplingPlugin
UpsamplingPlugin copied to clipboard
请问下trt.PluginField里如果type是trt.PluginFieldType.DIMS该怎么传data?
trafficstars
发现如果按照下面这样写貌似不太对,传进去之后,inputDims的nbDims会是一个随机数的样子:
inputDims_field = trt.PluginField("inputDims", np.array([16, 8, 8]), trt.PluginFieldType.DIMS)
Have you solve this problem?