gpt4 book ai didi

python - Tensorboard (PyTorch) add_graph 中的错误

转载 作者:行者123 更新时间:2023-12-03 23:49:17 25 4
gpt4 key购买 nike

我正在关注这个 Pytorch 的 Tensorboard 文档。
我有以下代码:

model = torchvision.models.resnet50(False)
writer.add_graph(model)
它引发以下错误:

_ = model(*args) # don't catch, just print the error message

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


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

最佳答案

我也有这个问题..

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

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

关于python - Tensorboard (PyTorch) add_graph 中的错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60021266/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com