torchstat

torchstat copied to clipboard

AttributeError: 'tuple' object has no attribute 'size'

1

np.array(output.size()[1:], dtype=np.int32)) AttributeError: 'tuple' object has no attribute 'size'

OpenAI-chn

AssertionError for the more-than-two-dimensional input of nn.Linear

6

Considering that the input of the linear layer is not necessarily only two-dimensional, I think these changes * change `assert len(inp.size()) == 2 and len(out.size()) == 2` to `assert len(inp.size())...

bighuang624

I think it's a amazing tool for CNN, I cann't wait to use it in CNN. Hope for torch.nn.functional very much, thanks for your great job!!!

Suncheng2019

For 3D Networks?

3

Hello! I want to check my 3D CNN model's GPU using with torchstat, but it seems not support inputs with more than 3D... assertionerror occured.. any solutions to solve this...

DISAPPEARED13

Not work for EfficientNet

3

lfangyu09

when i run the module，i got this error

4

File "static.py", line 10, in stat(model,(3,224,224)) File "/home/sk49/.local/lib/python3.6/site-packages/torchstat/statistics.py", line 71, in stat ms.show_report() File "/home/sk49/.local/lib/python3.6/site-packages/torchstat/statistics.py", line 64, in show_report collected_nodes = self._analyze_model() File "/home/sk49/.local/lib/python3.6/site-packages/torchstat/statistics.py", line 57, in _analyze_model model_hook =...

Re3write

How does torchstat compute total memory?

2

How does torchstat compute total memory for a model? For example, I use torchstat to analyze mobilenet and get the following results: ``` Total params: 4,231,976 -------------------------------------- Total memory: 38.48MB...

un-knight

‘Total MAdd’ and 'Total Flops' results should exchange with each other

12

MAdd means that the multiply and add operations are counted as 1 operation while FLOPs count them as 2. Should the results of 'Total MAdd' and 'Total Flops' exchange with...

Lukzin

not work in jupyter notebook?

1

can't import torchstat in jupyter's cell?

201528014227051

Some layer is not supported!

2

Thanks for your good job, however i found some layer can not be calculated. GroupNorm is not supported AdaptiveAvgPool2d is not supported Identity is not supported

Hengwei-Zhao96