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 {
public:
void Make() {
AddAttr<std::vector<std::string>>(
"epmap",
AddAttr<std::vector<std::string>>("epmap",
"(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"});
AddAttr<std::string>(
"dir", "(string, default '') indicate the folder checkpoint will use");

Loading…
Cancel
Save