我在关注这个皮托奇的。Tensorboard 文档。
我有以下代码。
model = torchvision.models.resnet50(False)
writer.add_graph(model)
它抛出了以下错误:
_ = model(*args) # 不要抓取，只需打印错误信息。
类型错误。ResNet对象参数*后必须是一个可迭代的对象，而不是NoneType。
我不知道我在这里做错了什么!
我也遇到了这个问题...
传递一个 input_to_model 不同的参数 None 解决了这个问题。但是，我觉得这应该是可选的
dataiter = iter(trainloader)
images, labels = dataiter.next()
writer.add_graph(model, images)