I’m following this Pytorch’s Tensorboard documentation.
I have the following code:
model = torchvision.models.resnet50(False)
writer.add_graph(model)

It throws the following error:

_ = model(*args)  # don’t catch, just print the error message
TypeError: ResNet object argument after * must be an iterable, not NoneType

I don’t know what I’m doing wrong here!