TrainingTools I want to write the different data sets to tensorflow2.0's tf.dataset which read from tfrecord.