Ошибка в add_graph в Tensorboard (PyTorch)

Question

Я слежу за Tensorboard этой Pytorch документацией.

У меня есть следующий код:

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

Выдает следующую ошибку:

_ = model (* args) # не перехватывать, просто напечатать сообщение об ошибке

TypeError: Re sNet аргумент объекта после * должен быть повторяемым, а не NoneType

Я не знаю, что я здесь делаю не так!

Answer 1

У меня тоже была эта проблема ..

Передача параметра input_to_model, отличного от None, решила проблему. Тем не менее, я думаю, что это не обязательно

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