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 d3b7690f21.
|
2023-03-14 16:47:13 +08:00 |
|
gaotingquan
|
d3b7690f21
|
add the amp decorator
|
2023-03-10 16:56:55 +08:00 |