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.