未验证 提交 d62dea9a 编写于 作者: K Kai Song 提交者: GitHub

[Custom Device]add lr_ratio support of adamw for custom device (#56386)

* [Custom Device]add lr_ratio support of adamw for mlu

* fix

* fix
上级 d084a236
...@@ -182,6 +182,8 @@ class AdamW(Optimizer): ...@@ -182,6 +182,8 @@ class AdamW(Optimizer):
if ( if (
not core.is_compiled_with_cuda() not core.is_compiled_with_cuda()
and not core.is_compiled_with_xpu() and not core.is_compiled_with_xpu()
and paddle.device.get_device().split(":")[0]
not in paddle.device.get_all_custom_device_type()
): ):
raise NotImplementedError("'lr_ratio' is unimplemented in CPU.") raise NotImplementedError("'lr_ratio' is unimplemented in CPU.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册