Tensorboard - ValueError：解包的值太多（预期 2）

Posted 2023-03-27

【中文标题】Tensorboard - ValueError：解包的值太多（预期 2）

【英文标题】：Tensorboard - ValueError: too many values to unpack (expected 2)

【发布时间】：2021-07-05 12:20:19

【问题描述】：

我尝试使用 tensorboard 来可视化模型。我正在关注 pytorch.org tutorial。这是数据加载器的代码。

writer_train = SummaryWriter('runs/training')

train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=config.train_batch_size, shuffle=True,
                                           num_workers=config.num_workers, pin_memory=True)

images, labels = next(iter(train_loader))
writer_train.graph_model(light_net, images)

我在 iter 行中遇到了这个错误。

图像，标签 = next(iter(train_loader)) ValueError：要解压的值太多（预期为 2）

我已经调试了代码并找到了这个。

【问题讨论】：

你的数据集是什么形状的？试试iter(train_loader).next()。

@iacob 谢谢你成功了！

【参考方案1】：

该错误可能是由于使用了内置函数而不是train_loader 对象的.next() 方法引起的。

next() 和iter() 是Python 中的内置方法。请参阅文档 iter 和 next。

在教程中显示如下

# get some random training images
dataiter = iter(trainloader)
images, labels = dataiter.next()

它使用next() 方法将值解压缩到两个变量中。这与您对next(iter(train_loader)) 的用法不同。按照所示的方式进行操作，它应该可以解决您的问题。

【讨论】：

谢谢你，当我尝试不带标签时它起作用了。 "图像 = dataiter.next()"