add SetParam

refactor_docs
Helin Wang 9 years ago
parent 0ed68da9b3
commit cd978b70f4

@ -57,9 +57,15 @@ type UpdateRule struct {
// closing the send channel.
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
// 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.
func (*PServerClient) GetParam(names []string) ([]Parameter, error)

Loading…
Cancel
Save