AI智能
改变未来

TypeError: batch() got an unexpected keyword argument drop_remainder

报错:

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)
赞(0) 打赏
未经允许不得转载:爱站程序员基地 » TypeError: batch() got an unexpected keyword argument drop_remainder