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

[Fix] Fix mdcn trt (#2075)

parent 5e6ede99
......@@ -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];
}
......
......@@ -254,7 +254,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin(
}
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];
}
......
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