自定义dataloader里的batch_sampler,你就不需要再做padding了!前言正文完整代码后记
前言今天看代码,居然已经水了三篇…这是第四篇。在说batch_sampler之前,这里不要脸的放一下之前关于dataset和dataloader的博客正文我在下面的代码中第一次看到了batch_sampler,于是好奇去看看这个FixedLengthBatchSampler到底在干嘛。iter(self)首先它放到dataloader里,肯定是要迭代生成一些东西,所以先去看它的__iter__()然后发现它做了两件事情...