提交 4388ce11 编写于 作者: T tangwei12

checkpoint notify op optimized

上级 7fae9e0a
...@@ -55,10 +55,9 @@ class CheckpointNotifyOp : public framework::OperatorBase { ...@@ -55,10 +55,9 @@ class CheckpointNotifyOp : public framework::OperatorBase {
class CheckpointNotifyOpMaker : public framework::OpProtoAndCheckerMaker { class CheckpointNotifyOpMaker : public framework::OpProtoAndCheckerMaker {
public: public:
void Make() { void Make() {
AddAttr<std::vector<std::string>>( AddAttr<std::vector<std::string>>("epmap",
"epmap", "(string vector, default 127.0.0.1:6164)"
"(string vector, default 127.0.0.1:6164)" "Parameter Server endpoints in the order")
"Server endpoints in the order of input variables for mapping")
.SetDefault({"127.0.0.1:6164"}); .SetDefault({"127.0.0.1:6164"});
AddAttr<std::string>( AddAttr<std::string>(
"dir", "(string, default '') indicate the folder checkpoint will use"); "dir", "(string, default '') indicate the folder checkpoint will use");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册