模型开发接口
推理部署接口
API 相关说明
具有张量融合优化的 Allreduce 回调函数。
reduce_method (str) – 归并梯度的方法。
str
group (Group) – 通信组。
Group
backend (Optional[str]) – 在allreduce中覆盖分布式后端
Optional
上一页
make_allreduce_cb
下一页
megengine.distributed.helper.param_pack_split