From cd978b70f4162e454a1c1bbfa67a52250469a4d4 Mon Sep 17 00:00:00 2001 From: Helin Wang Date: Tue, 9 May 2017 18:02:51 -0700 Subject: [PATCH] add SetParam --- doc/design/cluster_train/trainer.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/doc/design/cluster_train/trainer.md b/doc/design/cluster_train/trainer.md index c9c090dc984..bcb4a9c09dc 100644 --- a/doc/design/cluster_train/trainer.md +++ b/doc/design/cluster_train/trainer.md @@ -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) -- GitLab