触摸板(PyTorch)ADD_GRAPH中出错 [英] Error in Tensorboard's(PyTorch) add_graph
本文介绍了触摸板(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)
这篇关于触摸板(PyTorch)ADD_GRAPH中出错的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！
查看全文
