使用dataset进行数据管道化处理时,通常我们会加上batch(batch_size)来获取批量样本。这里有个容易忽视的点,batch本身还提供了一个参数drop_remaindar,用于标示是否对于最后一个batch如果数据量达不到batch_size时保留还是抛弃。
转载于:https://juejin.im/post/5d0203e2e51d45773e418a63
使用dataset进行数据管道化处理时,通常我们会加上batch(batch_size)来获取批量样本。这里有个容易忽视的点,batch本身还提供了一个参数drop_remaindar,用于标示是否对于最后一个batch如果数据量达不到batch_size时保留还是抛弃。
转载于:https://juejin.im/post/5d0203e2e51d45773e418a63