fast-reid/fastreid/engine
liaoxingyu 7e652fea2a feat: Add contiguous parameters support
Support contiguous parameters to train faster. It can split parameters into different contiguous groups by freeze_layer, lr and weight decay.
2021-07-05 11:10:37 +08:00
..
__init__.py finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
defaults.py feat: Add contiguous parameters support 2021-07-05 11:10:37 +08:00
hooks.py fix lr scheduler warning when amp training 2021-06-02 16:35:46 +08:00
launch.py onnx/trt support 2020-07-29 17:43:39 +08:00
train_loop.py feat: Add contiguous parameters support 2021-07-05 11:10:37 +08:00