CollieBatchSampler¶
- class CollieBatchSampler(sampler, batch_size, last_batch='normal')[源代码]¶
Batch Sampler。在最后一个 batch 样本数目不足一个
batch size时可以选择 不处理(normal)、丢弃(drop)或从头补齐(fill)。- 参数:
sampler –
batch_size –
last_batch (default:
'normal') –当最后一个 batch 样本数不足一个
batch_size时的处理方式'normal'- 不进行任何特殊处理。'drop'- 丢弃最后一个 batch。'fill'- 将最后一个 batch 补齐到batch_size大小。