Tensorboard (PyTorch) add_graph 中的錯誤

Question

我正在關注這個 Pytorch 的Tensorboard文檔。

我有以下代碼：

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

它引發以下錯誤：

_ = model(*args) # 不捕捉，只打印錯誤信息

TypeError: * 之后的 ResNet 對象參數必須是可迭代的，而不是 NoneType

我不知道我在這里做錯了什么！

Answer 1

我也有這個問題..

傳遞一個不同於None的input_to_model參數解決了這個問題。 但是，我認為它應該是可選的

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