У меня есть рабочий класс, который я использую для распределения вычислений по нескольким...

TensorFlow Estimator прост в использовании для распределенного обучения со стратегией сервера...