前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: zhangting2020
The usage of init_on_cpu does not take effect because it only works for some Ops with force_cpu Attr. For more information, please refer to this PR in models repo.
init_on_cpu
force_cpu