Documentation
¶
Index ¶
- type Client
- func (c *Client) BeginInitParams() (bool, error)
- func (c *Client) FinishInitParams() error
- func (c *Client) GetParams(names []string) ([]pserver.Parameter, error)
- func (c *Client) InitParam(paramWithConfigs pserver.ParameterWithConfig) error
- func (c *Client) SendGrads(grads []pserver.Gradient) error
- type Etcd
- type Lister
- type Selector
- type Server
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Client ¶
type Client struct {
// contains filtered or unexported fields
}
Client is the client to parameter servers.
func (*Client) BeginInitParams ¶
BeginInitParams begins to initialize parameters on parameter servers.
BeginInitParams will be called from multiple trainers, only one trainer will be selected to initialize the parameters on parameter servers. Other trainers will be blocked until the initialization is done, and they need to get the initialized parameters from parameter servers using GetParams.
func (*Client) FinishInitParams ¶
FinishInitParams tells parameter servers client has sent all parameters to parameter servers as initialization.
type Etcd ¶
type Etcd struct {
// contains filtered or unexported fields
}
Etcd is used by pserver client that is a part of trainer process. TODO: 1. add watcher to watch the change state of pservers.
type Lister ¶
type Lister interface {
List() []Server
}
Lister lists currently available parameter servers.
type Selector ¶
type Selector interface {
// Select selects if the client should initialize parameter servers.
Select() (bool, error)
// Done indicates the initialization process is done.
Done() error
}
Selector selects if the client should initialize parameters and reports the initialization process done.
Source Files
Directories