【Tensorflow2.0】去掉不足一个batch的多余数据
需要在数据的的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)
转载
2021-12-13 09:41:07 ·
618 阅读 ·
0 评论