transformers
Extend Trainer to enable Ascend NPU to use the fused Adamw optimizer when training
#26194
Merged

Extend Trainer to enable Ascend NPU to use the fused Adamw optimizer when training #26194

ji-huazhong
ji-huazhong ji-huazhong force pushed 2 years ago
ji-huazhong ji-huazhong force pushed 2 years ago
ji-huazhong Extend Trainer to enable Ascend NPU to use the fused Adamw optimizer …
0f4c7388
ji-huazhong ji-huazhong force pushed to 0f4c7388 2 years ago
ji-huazhong
ji-huazhong
muellerzr
muellerzr approved these changes on 2023-10-04
muellerzr muellerzr requested a review from ArthurZucker ArthurZucker 2 years ago
LysandreJik
LysandreJik approved these changes on 2023-10-04
LysandreJik LysandreJik merged 4fdf47cd into main 2 years ago
HuggingFaceDocBuilderDev
ji-huazhong ji-huazhong deleted the npu_fuse_optim branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone