mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/3240/head^2
huangyuan64 2025-02-10 12:20:10 +08:00 committed by GitHub
parent 1c11e0c004
commit 53e6b94ca7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp
View File

@ -53,7 +53,7 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois,
.Output(grad_offset) .Output(grad_offset)
.Attr("output_size", output_size) .Attr("output_size", output_size)
.Attr("spatial_scale", spatial_scale) .Attr("spatial_scale", spatial_scale)
.Attr("sample_ratio", sampling_ratio_) .Attr("sampling_ratio", sampling_ratio_)
.Attr("gamma", gamma) .Attr("gamma", gamma)
.Run(); .Run();
} }