触摸板(PyTorch)ADD_GRAPH中出错

问题描述

我正在遵循此Pytorch的Tensorboard文档。

我有以下代码：

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

它引发以下错误：

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

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

我不知道我做错了什么！

---

解决方案

我也遇到过这个问题..

传递与None不同的input_to_model参数解决了该问题。不过，我觉得应该是可选的

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