torchstat
torchstat copied to clipboard
The input shape
I think this tool may be quite useful, but there are still something not convenient for me... I want to get the data of my model, but since my model's input is (1, x, 1024), and the input of the stat() seems not to be friendly to me. Is there anyone else also facing this problem? Can anyone help me with this?
I have met the same problem, my data is point cloud
I also met with this problem. What does the input really mean?