使用 Horovod 进行 Keras 模型的数据并行训练

本教程演示如何使用 MPIJob 对 Keras 模型进行多工作器同步训练(使用 horovod.tensorflow.keras 模块)。

运行示例

请按照使用方法准备环境,然后前往本教程对应的示例,参照其 README 文档运行。

检查训练日志和指标

训练开始后,进入模型构建控制台的 Job 页面,可以看到名为 keras-mnist-mpijob 的 MPIJob 正在运行:

running

点击其名称进入详情页面,可以看到刚才创建的 MPIJob 的基本信息、状况信息和事件信息:

details

点击上方的副本标签页,查看 MPIJob 的 Pod 信息:

replicas

点击副本右侧的更多按钮 > 日志以查看训练脚本执行过程中的日志输出:

view-log

点击上方的指标标签页,查看 MPIJob 运行过程中使用集群计算资源、网络资源和存储资源的情况:

replicas

一段时间之后,MPIJob 的状态变为 Succeeded,表示训练成功完成:

done

若 MPIJob 在运行过程中出错,其状态会变为 Error,并在事件信息和 Pod 信息部分显示错误信息,此时需要根据给出的错误信息进行问题排查。