DCNv2
1.0.0
./make.sh # build
python testcpu.py # run examples and gradient check on cpu
python testcuda.py # run examples and gradient check on gpu 现在,主分支适用于Pytorch 1.x,您可以用,然后将其切换回Pytorch 0.4,
git checkout pytorch_0.4这是对官方可变形convnets的改编。
更新:所有梯度检查通过均具有双重精度。
另一个问题是,它增加了RuntimeError: Backward is not reentrant 。但是,错误非常小(float <1e-7 <1e-15 ,double)可能不是一个严重的问题(?)
如果您有任何评论,请发布问题或公关。