input_size can't change?

pcjiang1998 / torchpf

torchpf: A Tool for Pytorch Model Analyzer.

MIT License

5 stars 1 forks source link

input_size can't change? #1

Open AI-dsm opened 2 years ago

AI-dsm commented 2 years ago

Same code as 3 years ago!! "assert isinstance(input_size, (tuple, list)) and len(input_size) == 3" and subsequent code make us input very dull, I want input =(8,3,16,112,112) for 3DCNN, what can I do？Thank you for your answer

pcjiang1998 commented 2 years ago

It will be updated very soon. Some computing tools will be added for 3D modules.