I loaded a custom PyTorch model and I want to find out its input shape. Something like this:
model.input_shape
Is it possible to get this information?
Update: print() and summary() don't show this model's input shape, so they are not what I'm looking for.
from How to find out PyTorch model's input shape?
No comments:
Post a Comment