From 53e6b94ca744aaa8bda5807e7b256c40599f311d Mon Sep 17 00:00:00 2001 From: huangyuan64 Date: Mon, 10 Feb 2025 12:20:10 +0800 Subject: [PATCH] fix the npu deform roi pool param name error (#3235) --- mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp b/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp index 074e52d4f4..42de978e88 100644 --- a/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp +++ b/mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp @@ -53,7 +53,7 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois, .Output(grad_offset) .Attr("output_size", output_size) .Attr("spatial_scale", spatial_scale) - .Attr("sample_ratio", sampling_ratio_) + .Attr("sampling_ratio", sampling_ratio_) .Attr("gamma", gamma) .Run(); }