TylerYep/torchinfo

Issues

• Add the "groups" column to show number of channel groups.

  #319 opened 23 days ago by andravin
  3

• Issue with `torchinfo.summary()` Failing on Models with `torch.distributions.Categorical`

  #329 opened a month ago by Naeemkh
  0

• networks with ModuleList

  #327 opened a month ago by caiuspetronius
  0

• unsupported operand type(s) for +: 'int' and 'str'

  #268 opened a year ago by PANFEI-CHENG
  2

• TypeError: unsupported operand type(s) for +: 'int' and 'NoneType'

  #258 opened a year ago by YasmineXXX
  1

• Investigating overestimation of total parameter counts for multiple models

  #322 opened 2 months ago by 6DammK9
  1

• The parameter of stable diffusion unet is wrong

  #289 opened a year ago by FlyHighest
  1

• I got different results using thop and torchinfo

  #312 opened 6 months ago by zz-2024
  2

• Running `torchinfo.summary` on YOLO from `ultralytics` package?

  #309 opened 6 months ago by ego-thales
  1

• torchinfo.summary should not change the device of model

  #287 opened a year ago by gitxw
  2

• multi-parameters forward function doesnt work

  #317 opened 4 months ago by IKetchup
  1

• Add Bytes and KB Units for small PyTorch models

  #315 opened 4 months ago by halsimov
  1

• Summary not working with Model made out of submodules

  #313 opened 6 months ago by RHTNT
  1

• Error occurred while passing Empty-Vectors as parameters

  #311 opened 6 months ago by Liuqh12
  0

• Detecting dynamic flow?

  #310 opened 6 months ago by eliegoudout
  2

• The total number of parameters is different with huggingface model card.

  #303 opened 9 months ago by ErfanMoosaviMonazzah
  0

• Nested module not printing the output shape

  #299 opened 9 months ago by sup3rgiu
  5

• summary not working with a custom pre-processing layer

  #282 opened a year ago by jaroslawjanas
  2

• How to print only the: 'Estimated Total Size (MB):'

  #301 opened 10 months ago by aymuos15
  1

• Using batch_first=True for torch native transformer blocks seem to result in non-complete summaries

  #300 opened 10 months ago by cwestergren
  0

• How to get the summary if the model output consists of int and str?

  #280 opened a year ago by simon5u
  1

• Show summary up to error

  #288 opened a year ago by R-N
  0

• Fix bogus 1.64 git tag

  #286 opened a year ago by mweinelt
  1

• PyTorch Compile

  #277 opened a year ago by mohsenhariri
  0

• mat1 and mat2 shapes cannot be multiplied

  #269 opened a year ago by yash2002vardhan
  3

• Can I use summary with tuple input? (len(inputs) = 3)

  #272 opened a year ago by AppleJoker94
  2

• Misidentification of recursiveness (?)

  #262 opened a year ago by kswannet
  1

• Can torchinfo support BEVFusion (https://github.com/mit-han-lab/bevfusion) ?

  #267 opened a year ago by dpan817
  2

• verbose=2 shows weights twice

  #247 opened 2 years ago by matthijsvk
  1

• Why no input shape?

  #259 opened a year ago by xhxlab
  3

• Bug parsing bias argument

  #261 opened a year ago by amandalucasp
  0

• ERROR

  #252 opened a year ago by neverstoplearn
  4

• Support forward with multiple arguments

  #251 opened 2 years ago by joaolcguerreiro
  4

• get_total_memory_used fails to handle list of str

  #254 opened a year ago by minostauros
  1

• MPS support

  #253 opened 2 years ago by joaolcguerreiro
  2

• Write/Read Operation counter

  #246 opened 2 years ago by qmpzzpmq
  0

• TypedStorage is deprecated

  #245 opened 2 years ago by joaolcguerreiro
  1

• Error when getting summary of T5 model: Expected tensor for argument #1 'indices' to have one of the following scalar types

  #244 opened 2 years ago by anentropic
  4

• Summary does not show sizes for all elements using module that returns List[torch.tensor]

  #234 opened 2 years ago by xavierjimenezp
  0

• Inaccurate number of parameters at output layer

  #243 opened 2 years ago by John-CYHui
  0

• print model summary to latex

  #241 opened 2 years ago by yangtzech
  4

• pytorch2.0 warning

  #238 opened 2 years ago by YoRHazzz
  1

• Inaccurate Mult-Adds Estimation for Transformers

  #226 opened 2 years ago by Yiming-M
  3

• Error with model parallelism

  #224 opened 2 years ago by clessig
  3

• After calling summary(), model on CPU is pushed to GPU

  #227 opened 2 years ago by ZhilinGuo
  0

• Device changed to CPU when calling torchinfo.summary() when model is created on MPS for Mac Users

  #231 opened 2 years ago by frankcaoyun
  4

• Remove gradient calculation during inferencing (torch.no_grad)

  #230 opened 2 years ago by SinChee
  1

• Can we print summary on sequential model (RNN, LSTM, GRU)?

  #233 opened 2 years ago by jil818
  0

• Summary failed for transformers.GPT2Model

  #229 opened 2 years ago by apivovarov
  1

• Cannot assign str/repr from model.summary to a variable without 'workaround' in Colab

  #218 opened 2 years ago by Antsthebul
  5