RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.

分类: 文章 • 2024-11-24 09:09:27

原因：输入的图像数据的维度不完全一样(数据集图像大小不一)。

解决方法： transforms.Compose([
transforms.RandomResizedCrop(224),# 随机裁剪])

图片: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.