Skip to main content
 首页 » 编程设计

python之如何使用 Tensorflow 的 Estimator API 进行分布式训练

2024年10月01日27arxive

documentation说:

You can run Estimators-based models on a local host or on a distributed multi-server environment without changing your model. Furthermore, you can run Estimators-based models on CPUs, GPUs, or TPUs without recoding your model.

是否有文档解释如何在分布式多服务器环境中运行 Estimator?

请您参考如下方法:

tf.estimator.train_and_evaluate 的 Tensorflow 文档 here解释了一种在分布式环境中运行 tf.estimator 的方法,它只需要适本地设置 TF_CONFIG 环境变量。