torchview

torchview: visualize pytorch models

MIT License

Downloads
8.1K
Stars
803
Committers
2

Bot releases are hidden (Show)

torchview - - TorchView 0.2.6 Latest Release

Published by mert-kurttutan over 1 year ago

  • Fix readme format on pypi
  • Replace flake8 with ruff (only relevant for development issue)
torchview - TorchView 0.2.5

Published by mert-kurttutan over 1 year ago

  • Fixed the issue about isolated Tensor nodes
    • Issue: #73 , PR: #74
  • Complete coverage of tensor creation ops
    • PR: #74
torchview - Torchview 0.2.4

Published by mert-kurttutan almost 2 years ago

  • Fixed the bug for modules that changes mutable inputs, e.g. list of tensors PR
    • PR: #70 , Issue: #63
  • Very limited support for traced modules via #71 , see why here #66
  • #66 also hints at the solution for support for traced modules for those who want to implement and contribute the project
torchview - TorchView 0.2.3

Published by mert-kurttutan almost 2 years ago

  • Added support for change the direction of graph
  • Fixed issues for modules that create tensor during forward prop
torchview - TorchView 0.2.2

Published by mert-kurttutan almost 2 years ago

  • Support for more flexible architectures (Reuse of input, unused input tensors)
  • Increased support for Huggingface models including, bert, gpt2, detectron
torchview - TorchView 0.2.1

Published by mert-kurttutan almost 2 years ago

Increased support for some huggingface tokenizer

torchview - Torchview 0.2.0 Release

Published by mert-kurttutan almost 2 years ago

  • new display feature for nested modules ('expand_nested') very similar to keras plot_model
  • new test models
  • code cleaning
torchview - Initial Release

Published by mert-kurttutan almost 2 years ago