ndarray
autograd
loss
KVStore.
num_workers
Returns the number of worker nodes.
size – The number of worker nodes.
int