Issues
- 0
The parameter of stable diffusion unet is wrong
#289 opened by FlyHighest - 0
The total number of parameters is different with huggingface model card.
#303 opened by ErfanMoosaviMonazzah - 5
Nested module not printing the output shape
#299 opened by sup3rgiu - 2
- 1
How to print only the: 'Estimated Total Size (MB):'
#301 opened by aymuos15 - 0
Using batch_first=True for torch native transformer blocks seem to result in non-complete summaries
#300 opened by cwestergren - 1
- 0
Show summary up to error
#288 opened by R-N - 0
torchinfo.summary should not change the device of model
#287 opened by gitxw - 1
Fix bogus 1.64 git tag
#286 opened by mweinelt - 3
nn.ParameterList omitted again in v1.7.1
#190 opened by github-0-searcher - 0
PyTorch Compile
#277 opened by mohsenhariri - 3
mat1 and mat2 shapes cannot be multiplied
#269 opened by yash2002vardhan - 2
- 1
- 1
Misidentification of recursiveness (?)
#262 opened by kswannet - 2
- 1
verbose=2 shows weights twice
#247 opened by matthijsvk - 3
Why no input shape?
#259 opened by xhxlab - 0
Bug parsing bias argument
#261 opened by amandalucasp - 0
- 4
ERROR
#252 opened by neverstoplearn - 4
Support forward with multiple arguments
#251 opened by joaolcguerreiro - 1
get_total_memory_used fails to handle list of str
#254 opened by minostauros - 2
MPS support
#253 opened by joaolcguerreiro - 0
Write/Read Operation counter
#246 opened by qmpzzpmq - 1
TypedStorage is deprecated
#245 opened by joaolcguerreiro - 4
Error when getting summary of T5 model: Expected tensor for argument #1 'indices' to have one of the following scalar types
#244 opened by anentropic - 0
Summary does not show sizes for all elements using module that returns List[torch.tensor]
#234 opened by xavierjimenezp - 0
Inaccurate number of parameters at output layer
#243 opened by John-CYHui - 4
print model summary to latex
#241 opened by yangtzech - 1
pytorch2.0 warning
#238 opened by YoRHazzz - 3
Inaccurate Mult-Adds Estimation for Transformers
#226 opened by Yiming-M - 3
Error with model parallelism
#224 opened by clessig - 0
After calling summary(), model on CPU is pushed to GPU
#227 opened by ZhilinGuo - 4
Device changed to CPU when calling torchinfo.summary() when model is created on MPS for Mac Users
#231 opened by frankcaoyun - 1
- 0
- 1
Summary failed for transformers.GPT2Model
#229 opened by apivovarov - 1
conflict with compressai
#215 opened by Freed-Wu - 2
torchinfo.summary() does not show internal layers of torch.nn.TransformerEncoderLayer (in version 1.7.1)
#213 opened by AlessandroMiola - 5
Cannot assign str/repr from model.summary to a variable without 'workaround' in Colab
#218 opened by Antsthebul - 2
incosistent behaviour on ptpython
#216 opened by Freed-Wu - 2
- 1
Using torchinfo changes model's output even though random seed is fixed.
#203 opened by Fanning-Zhang - 1
CPU vs GPU usage
#202 opened by abs-xyz - 3
- 4
- 4
CUDA error: Out of memory
#197 opened by bsridatta - 2
Resnet hierarchy get wrong
#194 opened by BigDevil82