Skip to content

[Bugs] Disable grad scaler in AmpOptimWrapper when dtype is bf16#1553

Open
HIT-cwh wants to merge 2 commits intoopen-mmlab:mainfrom
HIT-cwh:fix_bf16_grad_scaler
Open

[Bugs] Disable grad scaler in AmpOptimWrapper when dtype is bf16#1553
HIT-cwh wants to merge 2 commits intoopen-mmlab:mainfrom
HIT-cwh:fix_bf16_grad_scaler

Commits

Commits on Jun 7, 2024