liaoxingyu 8f8cbf9411 fix lr scheduler warning when amp training ...

Skip lr scheduler when this iteration creates NaN gradients

2021-06-02 16:35:46 +08:00

..

__init__.py

finish v0.2 ddp training

2020-07-06 16:57:43 +08:00

defaults.py

fix lr scheduler warning when amp training

2021-06-02 16:35:46 +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

remove apex dependency (#442)

2021-03-23 12:12:35 +08:00