torchinfo

torchinfo copied to clipboard

Wrong recursive tag and duplicated/unordered entries

2

**Describe the bug** `summary` wrongly reports modules as recursive and several entries are duplicated and not in any logical order. **To Reproduce** Steps to reproduce the behavior: 1. `conda create...

CappellatoAlessio

Summary doesn't give the number of parameters for a GCN

1

**Describe the bug** When the summary on the Graph Convolutional Networks model is called and the model input is provided, the output summary doesn't contain any numbers in the column...

imaspol

fails on model returning ndarray(s)

1

**Describe the bug** `torchinfo.summary()` fails on a model that returns numpy ndarray instead of tensors. **To Reproduce** first install https://mmpose.readthedocs.io ``` from pathlib import Path import mim import torchinfo import...

smidm

Memory estimation inconsistent with actual GPU memory utilization

7

**Describe the bug** Memory estimation inconsistent with actual GPU memory utilization **To Reproduce** - I am using a simple UNet with 2 layers (same as [here](https://github.com/niuchuangnn/noise2sim/blob/master/noise2sim/modeling/networks/unet.py#L89)). - The input size...

rodrigovimieiro

MAccs of some operations seem to be omitted

1

Hi, I find torchinfo seems omit the MAccs of some operations like `torch.mul` and `torch.matmul`. Is it some feature-to-be-added, or we do not need to count these operations when calculating...

cyh767

summary does not work for CLIP models (jit)

4

Trying to summarize a [CLIP](https://github.com/openai/CLIP) model results in an output where there are no values listed in the `Output Shape` column: ``` ========================================================================================== Layer (type:depth-idx) Output Shape Param # ==========================================================================================...

sgimmini

Memory estimates for recursive models are inconsistent with actual GPU usage

3

**Describe the bug** Memory estimates are inconsistent with actual GPU usage for recursive models **To Reproduce** Example code: ```python import torch from torch.nn import Sequential, Linear class MyModel(torch.nn.Module): def __init__(self,...

Ethan-yt

add show recursive option and add to default row setting

8

Related to issue #65

VinhLoiIT

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

1

**Describe the bug** A clear and concise description of what the bug is. **To Reproduce** Steps to reproduce the behavior: ``` from transformers import BertModel from torchinfo import summary bert_base_path...

lan2720

Error in computing Linear Layer Multiply adds

5

**Describe the bug** When the linear layer has a multidimensional input and output (shape with 3 dimensions or more) the computed multiple adds will be incorrect. **To Reproduce** Add line...

jlclemon