Giters
TylerYep
/
torchinfo
View model summaries in PyTorch!
Geek Repo:
Geek Repo
Github PK Tool:
Github PK Tool
Stargazers:
2315
Watchers:
19
Issues:
148
Forks:
106
TylerYep/torchinfo Issues
The parameter of stable diffusion unet is wrong
Updated
8 days ago
The total number of parameters is different with huggingface model card.
Updated
3 months ago
Nested module not printing the output shape
Closed
3 months ago
Comments count
5
summary not working with a custom pre-processing layer
Updated
3 months ago
Comments count
2
How to print only the: 'Estimated Total Size (MB):'
Closed
3 months ago
Comments count
1
Using batch_first=True for torch native transformer blocks seem to result in non-complete summaries
Updated
3 months ago
How to get the summary if the model output consists of int and str?
Updated
5 months ago
Comments count
1
Show summary up to error
Updated
5 months ago
torchinfo.summary should not change the device of model
Updated
5 months ago
Fix bogus 1.64 git tag
Closed
5 months ago
Comments count
1
nn.ParameterList omitted again in v1.7.1
Updated
7 months ago
Comments count
3
PyTorch Compile
Closed
7 months ago
mat1 and mat2 shapes cannot be multiplied
Updated
7 months ago
Comments count
3
Can I use summary with tuple input? (len(inputs) = 3)
Closed
8 months ago
Comments count
2
unsupported operand type(s) for +: 'int' and 'str'
Updated
8 months ago
Comments count
1
Misidentification of recursiveness (?)
Updated
8 months ago
Comments count
1
Can torchinfo support BEVFusion (https://github.com/mit-han-lab/bevfusion) ?
Updated
9 months ago
Comments count
2
verbose=2 shows weights twice
Updated
10 months ago
Comments count
1
Why no input shape?
Closed
10 months ago
Comments count
3
Bug parsing bias argument
Closed
10 months ago
TypeError: unsupported operand type(s) for +: 'int' and 'NoneType'
Updated
a year ago
ERROR
Closed
a year ago
Comments count
4
Support forward with multiple arguments
Updated
a year ago
Comments count
4
get_total_memory_used fails to handle list of str
Updated
a year ago
Comments count
1
MPS support
Closed
a year ago
Comments count
2
Write/Read Operation counter
Updated
a year ago
TypedStorage is deprecated
Closed
a year ago
Comments count
1
Error when getting summary of T5 model: Expected tensor for argument #1 'indices' to have one of the following scalar types
Closed
a year ago
Comments count
4
Summary does not show sizes for all elements using module that returns List[torch.tensor]
Updated
a year ago
Inaccurate number of parameters at output layer
Updated
a year ago
print model summary to latex
Updated
a year ago
Comments count
4
pytorch2.0 warning
Closed
a year ago
Comments count
1
Inaccurate Mult-Adds Estimation for Transformers
Updated
a year ago
Comments count
3
Error with model parallelism
Closed
a year ago
Comments count
3
After calling summary(), model on CPU is pushed to GPU
Closed
a year ago
Device changed to CPU when calling torchinfo.summary() when model is created on MPS for Mac Users
Closed
a year ago
Comments count
4
Remove gradient calculation during inferencing (torch.no_grad)
Closed
a year ago
Comments count
1
Can we print summary on sequential model (RNN, LSTM, GRU)?
Closed
a year ago
Summary failed for transformers.GPT2Model
Closed
a year ago
Comments count
1
conflict with compressai
Closed
a year ago
Comments count
1
torchinfo.summary() does not show internal layers of torch.nn.TransformerEncoderLayer (in version 1.7.1)
Closed
a year ago
Comments count
2
Cannot assign str/repr from model.summary to a variable without 'workaround' in Colab
Closed
a year ago
Comments count
5
incosistent behaviour on ptpython
Closed
a year ago
Comments count
2
Models on cuda are always moved to GPU 0 by torchinfo.summary
Closed
a year ago
Comments count
2
Using torchinfo changes model's output even though random seed is fixed.
Closed
a year ago
Comments count
1
CPU vs GPU usage
Closed
a year ago
Comments count
1
AttributeError when input type has no element_size() method
Updated
a year ago
Comments count
3
Percentage FLOPS or Multiply Adds - inspired by #199
Updated
a year ago
Comments count
4
CUDA error: Out of memory
Closed
2 years ago
Comments count
4
Resnet hierarchy get wrong
Closed
2 years ago
Comments count
2
Previous
Next