tests/test_ops/test_modulated_deform_conv.py · 4bd3b5027aee4e3bfde9e429a59f0eab3d341c3e · 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_modulated_deform_conv.py 4.46 KB

Replace test_modulated_deform_conv.py