Shortcuts

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