|
|
|
@ -56,9 +56,9 @@ class RPCClient {
|
|
|
|
|
virtual void AsyncSendFetchBarrier(const std::string& ep,
|
|
|
|
|
int64_t time_out = FLAGS_rpc_deadline) = 0;
|
|
|
|
|
|
|
|
|
|
virtual void AsyncCheckpointNotify(
|
|
|
|
|
const std::string& ep, const std::string& dir,
|
|
|
|
|
int64_t time_out = FLAGS_grpc_deadline) = 0;
|
|
|
|
|
virtual void AsyncCheckpointNotify(const std::string& ep,
|
|
|
|
|
const std::string& dir,
|
|
|
|
|
int64_t time_out = FLAGS_rpc_deadline) = 0;
|
|
|
|
|
|
|
|
|
|
// SendComplete tells all the server that current trainer have no more data
|
|
|
|
|
// to train, so that the pserver can reduce it's barrier count, and continue
|
|
|
|
|