如何解决pytorch dataloader中设置batch_size时出现的bug问题？

本文共计331个文字，预计阅读时间需要2分钟。

运行时错误：张量尺寸必须匹配，除了维度0外。在维度3中得到342和281。

1、

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333

2、

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1

bug1 修改方法是将图片resize 成一个尺度：

解决1：

如果用transforms来resize ，关于尺寸的transform就是transforms.Resize(input_size, interpolation=3)了。