提交 e6bc7bb0 编写于 作者: Y Yu Yang

Tuning code

上级 0ca28b85
......@@ -510,6 +510,11 @@ All parameter, weight, gradient are variables in Paddle.
return &self.GetLocalScopes();
},
py::return_value_policy::reference)
.def("local_scopes_len",
[](ParallelExecutor &self) { return self.GetLocalScopes().size(); })
.def("local_scope", [](ParallelExecutor &self,
size_t i) { return self.GetLocalScopes()[i]; },
py::return_value_policy::reference)
.def("run", &ParallelExecutor::Run);
BindRecordIOWriter(&m);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册