python/paddle/fluid/dygraph/amp/loss_scaler.py · e8809d995713084cf22051797856c01713ffa616 · Crayon鑫 / Paddle

由 zhangbo9674 提交于 9月 14, 2022

* support bfloat16 for amp_decorate

* add check_finite for bf16

* fix bug

* add ut

* add ut

* refine code

e8809d99

loss_scaler.py 21.4 KB

Crayon鑫 / Paddle 与 Fork 源项目一致