5 Commits

Author SHA1 Message Date
gaotingquan
14d06fb6bd support AMP.use_amp arg 2023-05-25 16:16:02 +08:00
gaotingquan
b0877289f4 disable promote kernel for amp training
compatible with paddle 2.5 and older version.
ref: https://github.com/PaddlePaddle/PaddleClas/pull/2798
2023-05-25 11:58:05 +08:00
gaotingquan
8b218b01ac refactor amp auto_cast context manager & loss scaler 2023-05-25 11:58:05 +08:00
Tingquan Gao
fd7ef078fe Revert "add the amp decorator"
This reverts commit d3b7690f2102e5911410176f857313479bcdd169.
2023-03-14 16:47:13 +08:00
gaotingquan
d3b7690f21 add the amp decorator 2023-03-10 16:56:55 +08:00