赞
踩
output = DCNv3.dcnv3_forward( RuntimeError: Not implemented on the CPU
这时我们只要在我们训练的py文件引入torch即可
import torch
并在default.yaml中,手动添加 device为你自己的显卡序列
如果用CLI训练的,添加cuda到命令行即可