fix the npu deform roi pool param name error (#3235)

2025-02-10 12:20:10 +08:00 · 2025-02-10 12:20:10 +08:00 · 53e6b94ca7
parent 1c11e0c004
commit 53e6b94ca7
1 changed files with 1 additions and 1 deletions
--- 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();
 }