使用 TensorFlow 2 进行分布式训练

spark-tensorflow-distributor 是 TensorFlow 中的开源本机包,可帮助用户在 Spark 群集上使用 TensorFlow 进行分布式训练。 它是在 tensorflow.distribute.Strategy(TensorFlow 2 中的主要功能之一)的基础上构建的。 有关详细的 API 文档,请参阅 docstrings。 有关分布式 TensorFlow 的常规文档,请参阅使用 TensorFlow 进行分布式训练

示例笔记本

使用 TensorFlow 2 进行分布式训练

获取笔记本