Skip to content

Changed to only run reshard hook if all gradients computed #713

Changed to only run reshard hook if all gradients computed

Changed to only run reshard hook if all gradients computed #713