echo245的博客

需要在数据的的batch中添加drop_remainder=True即可类似，如：train_dataset = tf.data.Dataset.from_tensor_slices((train_data, train_labels))train_dataset = train_dataset.shuffle(buffer_size=1024).batch(BATCH_SIZE, drop_remainder=True)