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

Question

我尝试使用张量板来可视化 model。 我正在关注 pytorch.org 教程。 这是数据加载器的代码。

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 行中遇到了这个错误。

图像，标签=下一个（iter（train_loader））ValueError：解包的值太多（预期2）

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

Answer 1

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

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

在教程中显示以下内容

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

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