New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add Resnet50 for bf16 mixed precision #5316

Open

AshburnLee wants to merge 1 commit into PaddlePaddle:dev-static from AshburnLee:resnet50-bf16

AshburnLee commented Jun 9, 2021 •

edited

PR功能

PR支持了GPU单卡 Resnet50 的bfloat16混合精度训练。

方法

添加参数use_amp_bf16 & use_pure_bf16 并设false为初始值。
在train.py中调用接口decorate_bf16()，并指定使用bfloat16精度的op，将其放入custom_bf16_list中。
添加执行bf16 混合精度训练的脚本Resnet50_bf16.sh


          Add Resnet50 for bf16 mixed precision

7d9aef8

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment