NPU对pytorch,想将检测模型在NPU上训练,存在编译MultiScaleDeformableAttention的需求。
然而,原dino模型https://github.com/IDEA-Research/DINO/tree/main/models/dino/ops/src
仅包含CPU版本和GPU版本:
是不是就真的无法解决了呢?
好在mmcv支持了NPU版本,可以通过编译NPU版本的mmcv实现(参考:https://mmcv.readthedocs.io/en/master/get_started/build.html):
https://github.com/open-mmlab/mmcvd
通过以上链接下载源码