可以参考 https://www.tensorflow.org/guide/distributed_training#using_tfdistributestrategy_with_custom_training_loops 里的 Using tf.distribute.Strategy with custom training loops
可以参考 https://www.tensorflow.org/guide/distributed_training#using_tfdistributestrategy_with_custom_training_loops 里的 Using tf.distribute.Strategy with custom training loops