热门

最新

红包

立Flag

投票

同城

发布
devfranoischollet
Dreamweaver183
1 年前
truedevfranoischollet

我见过的其他模型并行API都在模型代码中内联了分区注释，并使用自定义训练逻辑——这使得修改和维护分区配置变得具有挑战性。Keras 3极大地简化了这一过程。
1
24
  • 复制链接
  • 举报
下一条：
Keras 3 + JAX 使得通过数据并行和模型并行（变量分割）在任意多的设备上高效分布大型模型成为可能。它不需要修改您的模型代码或训练代码——您需要做的只是单独指定一个“LayoutMap”，它告诉框架如何分割或复制模型中的每个变量。