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

Tuning code

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