Extend Trainer to enable Ascend NPU to use the fused Adamw optimizer when training #26194
Extend Trainer to enable Ascend NPU to use the fused Adamw optimizer …
0f4c7388
muellerzr
approved these changes
on 2023-10-04
ji-huazhong
deleted the npu_fuse_optim branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub