Commit Graph

4 Commits (296fbea9893717b1a775e1319d884db1bd8d6d40)

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