From bdabdffcf95757f2c25b7c1be15dae02d54c34ea Mon Sep 17 00:00:00 2001 From: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com> Date: Tue, 15 Mar 2022 11:15:15 +0800 Subject: [PATCH] [Fix] Fix deform_conv inconsistent pybind (#1781) --- mmcv/ops/csrc/pytorch/pybind.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/pybind.cpp b/mmcv/ops/csrc/pytorch/pybind.cpp index 88f32b6be..064aabc07 100644 --- a/mmcv/ops/csrc/pytorch/pybind.cpp +++ b/mmcv/ops/csrc/pytorch/pybind.cpp @@ -470,21 +470,21 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { m.def("deform_conv_forward", &deform_conv_forward, "deform_conv_forward", py::arg("input"), py::arg("weight"), py::arg("offset"), py::arg("output"), py::arg("columns"), py::arg("ones"), py::arg("kW"), - py::arg("kH"), py::arg("dW"), py::arg("dH"), py::arg("padH"), - py::arg("padW"), py::arg("dilationW"), py::arg("dilationH"), + py::arg("kH"), py::arg("dW"), py::arg("dH"), py::arg("padW"), + py::arg("padH"), py::arg("dilationW"), py::arg("dilationH"), py::arg("group"), py::arg("deformable_group"), py::arg("im2col_step")); m.def("deform_conv_backward_input", &deform_conv_backward_input, "deform_conv_backward_input", py::arg("input"), py::arg("offset"), py::arg("gradOutput"), py::arg("gradInput"), py::arg("gradOffset"), py::arg("weight"), py::arg("columns"), py::arg("kW"), py::arg("kH"), - py::arg("dW"), py::arg("dH"), py::arg("padH"), py::arg("padW"), + py::arg("dW"), py::arg("dH"), py::arg("padW"), py::arg("padH"), py::arg("dilationW"), py::arg("dilationH"), py::arg("group"), py::arg("deformable_group"), py::arg("im2col_step")); m.def("deform_conv_backward_parameters", &deform_conv_backward_parameters, "deform_conv_backward_parameters", py::arg("input"), py::arg("offset"), py::arg("gradOutput"), py::arg("gradWeight"), py::arg("columns"), py::arg("ones"), py::arg("kW"), py::arg("kH"), py::arg("dW"), - py::arg("dH"), py::arg("padH"), py::arg("padW"), py::arg("dilationW"), + py::arg("dH"), py::arg("padW"), py::arg("padH"), py::arg("dilationW"), py::arg("dilationH"), py::arg("group"), py::arg("deformable_group"), py::arg("scale"), py::arg("im2col_step")); m.def("deform_roi_pool_forward", &deform_roi_pool_forward,