提交 cd978b70 编写于 作者: H Helin Wang

add SetParam

上级 0ed68da9
...@@ -57,9 +57,15 @@ type UpdateRule struct { ...@@ -57,9 +57,15 @@ type UpdateRule struct {
// closing the send channel. // closing the send channel.
func (*PServerClient) ParamInitChan() (send chan<- Parameter, err error) func (*PServerClient) ParamInitChan() (send chan<- Parameter, err error)
// SendGrad sends gradients to parameter servers. // SendGrad sends gradients to parameter servers for updating parameters.
func (*PServerClient) SendGrad(method UpdateMethod, grads []Gradient) error func (*PServerClient) SendGrad(method UpdateMethod, grads []Gradient) error
// SetParam sets parameters.
//
// SetParam can be used for the parameters that are not suitable for updating
// using gradients.
func (*PServerClient) SetParam(params []Paramter) error
// GetParam gets parameters from parameter servers. // GetParam gets parameters from parameter servers.
func (*PServerClient) GetParam(names []string) ([]Parameter, error) func (*PServerClient) GetParam(names []string) ([]Parameter, error)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册