pytorch 样本数据无法整除batchsize怎么办

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch 样本数据无法整除batchsize怎么办相关的知识，希望对你有一定的参考价值。

train_loader = DataLoader(dataset, batch_size=args.batch_size, shuffle=False, drop_last=True)

在加载数据集的时候，加入

drop_last=True

以上是关于pytorch 样本数据无法整除batchsize怎么办的主要内容，如果未能解决你的问题，请参考以下文章