想使用torch自带的dcn,代替代码中的DCN编译算子,请问大家如何下手?还是建议不修改
stealth0414 opened this issue · 1 comments
stealth0414 commented
感觉每次在新环境中编译DCN算子时会担心编译失败
HoVDuc commented
instead of dcn you can use torchvision.ops.DeformConv2d https://pytorch.org/vision/main/generated/torchvision.ops.DeformConv2d.html