From 3ca8f031f59f6ef5355bce263d6cd7dcd0ab7768 Mon Sep 17 00:00:00 2001 From: dflhw <595869386@qq.com> Date: Thu, 12 Jan 2023 19:29:36 +0800 Subject: [PATCH] Modify DeformableRoiPoolGrad's parameter location --- mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp b/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp index 0e9f2ee7ac..074e52d4f4 100644 --- a/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp +++ b/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp @@ -45,11 +45,11 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois, int64_t sampling_ratio_ = (int64_t)sampling_ratio; OpCommand cmd; cmd.Name("DeformableRoiPoolGrad") - .Input(grad_input) + .Input(grad_output) .Input(input) .Input(rois) .Input(offset) - .Output(grad_output) + .Output(grad_input) .Output(grad_offset) .Attr("output_size", output_size) .Attr("spatial_scale", spatial_scale)