我目前正在使用生成器来使用tf.data.Dataset.from_generator生成我的培训和验证数据集。build_dataset(self, batch_size=16, shuffle=16, validation=None): train_dataset = tf.data.Dataset.from_generatorbatch_size).prefetch(1) if validation is not None:
val_dataset = tf.data.Dataset.from_<em