报错:
Traceback (most recent call last):File \"train_mv2.py\", line 652, in <module>train = Trainer()File \"train_mv2.py\", line 276, in train_netif mode == 0:File \"train_mv2.py\", line 296, in train_orbbectrain_op = self.train_op(loss, opt)File \"F:\\depth-face\\load_data_slim.py\", line 85, in get_orbbec_batchdataset = dataset.batch(batch_size,drop_remainder= True)TypeError: batch() got an unexpected keyword argument \'drop_remainder\'
tf1.10之后的版本才支持这种方式
dataset = dataset.batch(batch_size,drop_remainder= True)
保留整数个batch size 后的剩余数据,
而之前的版本只能使用
tf.contrib.data.batch_and_drop_remainder(batch_size)