Guangchen Lin 4bd3b5027a

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

* 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()'

2021-05-23 17:42:59 +08:00

..

data

Add test util for checking stand-alone python scripts (#1007)

2021-05-10 22:15:35 +08:00

test_cnn

[Fix] fix generalized attention fp16 (#1036)

2021-05-23 15:16:27 +08:00

test_image

[Feature] Add cutout and lighting (#909)

2021-03-26 18:58:19 +08:00

test_ops

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

2021-05-23 17:42:59 +08:00

test_runner

add neptune.ai logger hook (#1025)

2021-05-23 15:28:21 +08:00

test_utils

Add test util for checking stand-alone python scripts (#1007)

2021-05-10 22:15:35 +08:00

test_video

[Refactor] remove the c implementation of flow_warp (#707)

2020-12-13 19:43:25 +08:00

test_arraymisc.py

Add copyright header (#171)

2020-01-10 13:34:42 +08:00

test_fileclient.py

add HTTP backend (#860)

2021-03-03 16:23:04 +08:00

test_fileio.py

Add copyright header (#171)

2020-01-10 13:34:42 +08:00

test_load_model_zoo.py

map_location for all (#826)

2021-02-10 12:23:39 +08:00

test_parallel.py

Add pytorch 1.6 to CI (#476)

2020-08-09 21:56:15 +08:00

test_visualization.py

Add copyright header (#171)

2020-01-10 13:34:42 +08:00