hi,
I’m new to PyTorch. I have a network that trains and runs ok, except that Tensorboard doesnt work fully. With the following lines-
image = torch.zeros((2, 3, args.image_size, args.image_size))
model(image)
writer.add_graph(model, image)
I get the error-
*** TypeError: forward() takes 2 positional arguments but 3 were given
which I dont understand becoz training itself works.
Any thought?
Thanks!