python - Tensorboard (PyTorch) add_graph 中的错误

Question

我正在关注这个 Pytorch 的Tensorboard文档。

我有以下代码：

model = torchvision.models.resnet50(False)
writer.add_graph(model)

它抛出以下错误：

_ = model(*args) # 不捕获，只打印错误信息

TypeError: * 之后的 ResNet 对象参数必须是可迭代的，而不是 NoneType

我不知道我在这里做错了什么！

Answer 1

我也有这个问题..

传递一个input_to_model不同于None解决问题的参数。但是，我虽然它应该是可选的

dataiter = iter(trainloader)
images, labels = dataiter.next()
writer.add_graph(model, images)