如何解决图像简单CNN模型中的‘ValueError: too many values to unpack (expected 2)’错误

这个错误的核心是你的train_dl（DataLoader）每次迭代输出的元素数量，和你试图解构的images, labels两个变量不匹配。解决步骤如下：

• 排查Dataset的返回值
  检查你定义的Dataset类的__getitem__方法，确认它的返回值数量。比如如果该方法返回了(图像, 标签, 额外信息)三个元素，那循环里只拆成两个变量就会报错。确保__getitem__只返回图像和标签两个元素，或者根据实际返回数量调整循环变量。

• 查看batch的实际结构
  用以下代码打印出第一个batch的元素数量和形状，明确数据结构：

  batch = next(iter(train_dl))
  print("Batch元素数量:", len(batch))
  print("每个元素的形状:", [item.shape for item in batch])
  

• 调整循环的变量解构
  根据上面的结果修改循环：

  • 如果batch只有1个元素（比如仅图像）：

    for batch in train_dl:
        print("Images shape: ", batch.shape)
        out = simple_model(batch)
        print("out shape: ", out.shape)
        break
    

  • 如果batch有3个元素（比如图像、标签、其他信息）：

    for images, labels, extra_info in train_dl:
        print("Images shape: ", images.shape)
        out = simple_model(images)
        print("out shape: ", out.shape)
        break
    

内容的提问来源于stack exchange，提问作者Maruthi manideep Gorla