Fix the dimension (#1117)

59ed0ddd · Jaemin Jung · GitHub · f7caa80f · 59ed0ddd
Unverified Commit 59ed0ddd authored Jun 23, 2021 by Jaemin Jung Committed by GitHub Jun 23, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp +1 -1

No files found.
--- a/mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
+++ b/mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
@@ -65,7 +65,7 @@ nvinfer1::DimsExprs ModulatedDeformableConvPluginDynamic::getOutputDimensions(
  nvinfer1::DimsExprs ret;
  ret.nbDims = 4;
  ret.d[0] = inputs[0].d[0];
-  ret.d[1] = inputs[2].d[0];
+  ret.d[1] = inputs[3].d[0];

  ret.d[2] = inputs[1].d[2];
  ret.d[3] = inputs[1].d[3];