Dataset与Dataloader学习记录

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 19 and 1 in dimension 1 at /pytorch/aten/src/TH/generic/THTensor.cpp:689

在使用DataLoader中，遇到了一个问题

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 19 and 1 in dimension 1 at /pytorch/aten/src/TH/generic/THTensor.cpp:689

我的初始DataLoader代码是：

dataset=mydata(label_path)
batch_iterator=iter(DataLoader(dataset,batchsize,shuffle=True,num_workers=num_workers))

batch_size为32，数据集的总长度为16098
报错说明：batchsize在纬度1上不匹配
查看其他人的博客，说原因有两个：
你输入的图像数据的维度不完全是一样的，比如是训练的数据有100组，其中99组是256×256，但有一组是384×384，这样会导致Py