Hello I am very new to this field and
dealing with Pytorch is difficult for me.
How can I visualize the architecture of a model?
I have tried the following:
and it throws an error that I cannot solve.
Hello I am very new to this field and
dealing with Pytorch is difficult for me.
How can I visualize the architecture of a model?
I have tried the following:
The error suggests your model is in training mode but should be in evaluation mode. So, can you put model.eval()
before the yhat = model(batch)
line, and see if that solves the error?
batch = next(iter(training_data))
model.eval()
yhat=model(batch)
Thank you for your reply. After your addition I received another error message.
Your variable img
isn’t a Tensor it’s a list, so check what img
exactly is. It could be a list of a single tensor, but you’ll need to check. You could try printing len(img)
to see how many items are in the list