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用です。Pytorch0.4に戻ることができます。
git checkout pytorch_0.4これは、公式の変形可能なコンブネットの適応です。
更新:すべての勾配チェックパスは、 2倍の精度でパスします。
別の問題は、 RuntimeError: Backward is not reentrant 。ただし、エラーは非常に小さい(フロートでは<1e-7 <1e-15の場合は2倍)、深刻な問題ではない可能性があります(?)
コメントがある場合は、問題またはPRを投稿してください。