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 大小。