Python Tensorboard-值错误：要解压缩的值太多（预期为2个）

我曾尝试使用tensorboard来可视化模型。我在关注pytorch.org。下面是dataloader的代码

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（列车装载机）） ValueError:要解压缩的值太多（应为2个）

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

---

该错误可能是由于使用了内置函数而不是

火车加载器

对象的

.next（）

方法造成的

next（）

和

iter（）

是

Python

中的内置方法。请参阅文档和

在本教程中，将显示以下内容

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

---

其中，它使用

next（）

方法将值解压缩到2个变量中。这与您使用的

next（iter（火车装载机））

不同。按照所示的方式进行操作，它将解决您的问题。

数据集的形状是什么？试试iter（火车装载机）。下一步（）@iacob谢谢你，它成功了！谢谢，当我尝试不贴标签时，它起作用了。“images=dataiter.next（）”