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)