4 Commits

Author SHA1 Message Date
liaoxingyu
53fed7451d feat: support amp training
Summary: support automatic mixed precision training #217
2020-09-02 18:03:12 +08:00
liaoxingyu
ac8409a7da updating for pytorch1.6 2020-08-20 15:51:41 +08:00
liaoxingyu
35794621cc remove addmm warning 2020-07-31 16:32:10 +08:00
liaoxingyu
16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00