liaoxingyu 2b65882447 change way of layer freezing ... Remove `find_unused_parameters` in DDP and add a new step function in optimizer for freezing backbone. It will accelerate training speed in this way.		2021-05-25 15:57:09 +08:00
..
__init__.py	finish v0.2 ddp training	2020-07-06 16:57:43 +08:00
defaults.py	change way of layer freezing	2021-05-25 15:57:09 +08:00
hooks.py	change way of layer freezing	2021-05-25 15:57:09 +08:00
launch.py	onnx/trt support	2020-07-29 17:43:39 +08:00
train_loop.py	remove apex dependency (#442)	2021-03-23 12:12:35 +08:00