Tensorboard (PyTorch) add_graph 中的错误答案

【问题标题】：Error in Tensorboard's(PyTorch) add_graphTensorboard (PyTorch) add_graph 中的错误
【发布时间】：2020-05-18 03:34:24
【问题描述】：

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

我有以下代码：

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

它会抛出以下错误：

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

TypeError: ResNet object argument after * must be an iterable, not NoneType

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

【问题讨论】：

【解决方案1】：

我也有这个问题。。

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

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

【讨论】：