snuspl / cruise

Cruise: A Distributed Machine Learning Framework with Automatic System Configuration
Apache License 2.0
26 stars 2 forks source link

Use multi-update API for pushing model gradients #1259

Closed wynot12 closed 7 years ago

wynot12 commented 7 years ago

Currently, dolphin pushes model gradients for keys one by one. It incurs much overhead.

Especially it becomes worse with larger number of keys and smaller size of mini-batch. Sometimes, sender queue becomes full and computation is blocked by push() call.

We can optimize it by using multi-update API that batching requests for a same block.