Skip to content

Commit ff4b4cf

Browse files
authored
Merge pull request #2654 from dzhwinter/client_gradient
"fix client send empty gradients bug"
2 parents 95cda82 + c5e8e27 commit ff4b4cf

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

go/pserver/client.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package pserver
22

33
import (
4+
"errors"
45
"hash/fnv"
56
"sort"
67
"time"
@@ -123,6 +124,9 @@ func (c *Client) FinishInitParams() error {
123124
// SendGrads sends gradients to parameter servers for updating
124125
// parameters.
125126
func (c *Client) SendGrads(grads []Gradient) error {
127+
if len(grads) == 0 {
128+
return errors.New("no gradient received")
129+
}
126130
errCh := make(chan error, len(grads))
127131
for _, g := range grads {
128132
go func(g Gradient) {

0 commit comments

Comments
 (0)