Qwen

/

Qwen2.5-0.5B

like 267

Follow

Qwen 30.8k

Hello,

Is it possible to have the model input shape?
For example, you can get any TFLite model its input and output shape by running get_input_details()
How can I get those infos on the PyTorch model?
My goal is to compare the PyTorch input/output shapes to the TFLite input/ouput shapes

Thank you!