[AMP] support master_grad for amp training (#52235)
* support set master_grad * move register_hook to auto_cast * update unittest * fix fp16 test * update for review comments
Showing
test/amp/test_amp_master_grad.py
0 → 100644
想要评论请 注册 或 登录