Wenwei Zhang ed2887bb72

Support to specify LR of DCN's conv_offset (#344) ...

* Support to specify LR of DCN's conv_offset

* Resolve comments & add unit test

* Resolve formats

* Fix CI for DCN

* Mock DCN when cpu only

* Use mock for cpu testing

* Fix docstring and support ModulatedDCN

* set offset_lr_mult as dcn's arguments, link CU-49u01p

* fix lr bug

* fall back to set LR in constructor

* resolve comments

2020-09-25 20:51:58 +08:00

..

test_checkpoint.py

Remove all module wrapper's module when saving checkpoint (#399)

2020-07-08 23:20:22 +08:00

test_dist_utils.py

Specifying distributed training port in os.environ when training with slurm (#362)

2020-06-20 00:49:44 +08:00

test_fp16.py

Move unit tests to specific folders (#520)

2020-08-25 20:11:56 +08:00

test_hooks.py

Add runner builder (#570)

2020-09-25 10:25:29 +08:00

test_optimizer.py

Support to specify LR of DCN's conv_offset (#344)

2020-09-25 20:51:58 +08:00

test_runner.py

Add runner builder (#570)

2020-09-25 10:25:29 +08:00

test_utils.py

add unittest for set_random_seed (#376)

2020-07-02 00:13:04 +08:00