提交 5bf0c5bd 编写于 作者: G guosheng

Change back to load to CPU.

上级 ed08f236
......@@ -198,10 +198,7 @@ class StaticGraphAdapter(object):
assert param_state, "failed to load parameters, please check path"
if self._executor is None:
# TODO: loading to CPU seems to some transform error, and only
# the first step get the right result
# executor = fluid.Executor(fluid.CPUPlace())._default_executor
executor = fluid.Executor(fluid.CUDAPlace(0))._default_executor
executor = fluid.Executor(fluid.CPUPlace())._default_executor
else:
executor = self._executor._default_executor
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册