Skip to content

GitLab

Unverified Commit d71d067d authored by q.yao's avatar q.yao Committed by GitHub
Browse files

[Fix] Fix mdcn trt (#2075)

parent 5e6ede99
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/ops/csrc/tensorrt/plugins/trt_deform_conv.cpp
View file @ d71d067d
...@@ -282,7 +282,7 @@ nvinfer1::IPluginV2 *DeformableConvPluginDynamicCreator::createPlugin( ...@@ -282,7 +282,7 @@ nvinfer1::IPluginV2 *DeformableConvPluginDynamicCreator::createPlugin(
} }
} }
if (field_name.compare("deformable_group") == 0) { if (field_name.compare("deform_groups") == 0) {
deformableGroup = static_cast<const int *>(fc->fields[i].data)[0]; deformableGroup = static_cast<const int *>(fc->fields[i].data)[0];
} }
......
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
View file @ d71d067d
...@@ -254,7 +254,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin( ...@@ -254,7 +254,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin(
} }
std::string field_name(fc->fields[i].name); std::string field_name(fc->fields[i].name);
if (field_name.compare("deformable_group") == 0) { if (field_name.compare("deform_groups") == 0) {
deformableGroup = static_cast<const int *>(fc->fields[i].data)[0]; deformableGroup = static_cast<const int *>(fc->fields[i].data)[0];
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment