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.
and it throws an error that I cannot solve.
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.
What does “expecting the last two dimensions of the Tensor to be H and W instead got {img.shape[-2:]}” mean?
And how can I fulfill the expectation?
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