[sgd] Use target label count as training batch size (#16400)
[sgd] Use target label count as training batch size (#16400)


pcmoritz/ray-1,ray-project/ray,pcmoritz/ray-1,pcmoritz/ray-1,ray-project/ray,ray-project/ray,ray-project/ray,pcmoritz/ray-1,ray-project/ray,pcmoritz/ray-1,pcmoritz/ray-1,pcmoritz/ray-1,ray-project/ray,pcmoritz/ray-1,ray-project/ray,ray-project/ray