tests/test_ops/test_deform_conv.py · 4bd3b5027aee4e3bfde9e429a59f0eab3d341c3e · OpenDAS / MMCV

"vscode:/vscode.git/clone" did not exist on "cbfbab3f9800ef3f013949e1a744f8716c259d72"

[Fix] Support amp (pytorch >= 1.6.0) on DCN and DCNv2/ Add unit tests on DCN/DCNv2 amp (#1029) · 4bd3b502

Guangchen Lin authored May 23, 2021

* fix fp16 bug on DCNv2

* support fp16 on DCN/DCNv2 when pytorch >= '1.6.0'

* add comment

* Modified the comments

* Unified the usages of '.to()' and '.type_as()'

4bd3b502

test_deform_conv.py 6.15 KB

Replace test_deform_conv.py