checkpoint notify op optimized

port
tangwei12 7 years ago
parent 7fae9e0a7b
commit 4388ce112e

@ -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)"
"Server endpoints in the order of input variables for mapping") "Parameter Server endpoints in the order")
.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");

Loading…
Cancel
Save