TensorFlow 分布式训练

加一行dataset = dataset.prefetch(tf.data.AUTOTUNE)