tests/test_ops/test_deform_conv.py · 21845db455de64ca05878f990a80e9c78d0c33a4 · OpenDAS / MMCV

[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