torchinfo

View model summaries in PyTorch!

MIT License

Downloads
245.2K
Stars
2.5K
Committers
21
View Code on GitHub View on X
Ecosystems: Keras, PyTorch, Python

Commit Statistics

Past Year

All Time

Total Commits
20
382
Total Committers
3
23
Avg. Commits Per Committer
6.67
16.61
Bot Commits
12
91

Issue Statistics

Past Year

All Time

Total Pull Requests
25
86
Merged Pull Requests
19
72
Total Issues
20
82
Time to Close Issues
6 days
about 1 month
Package Rankings
Top 1.72% on Pypi.org
Top 6.59% on Proxy.golang.org
Top 14.68% on Conda-forge.org
Badges
Extracted from project README
Python 3.8+ PyPI version Conda version Build Status pre-commit.ci status GitHub license codecov Downloads
Related Projects
ssd_keras

A Keras port of Single Shot MultiBox Detector

02 Apr 2017 1,859
segmentation_training_pipeline

Research Pipeline for image masking/segmentation in Keras

23 Oct 2018 53
torchkeras

Pytorch❤️ Keras 😋😋

21 Jun 2020 1,715
pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

23 Apr 2018 4,007
keras-xla-benchmarks

Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.

28 May 2023 34
deepvision

PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/impleme...

31 Jan 2023 33
Keras-OneClassAnomalyDetection

[5 FPS - 150 FPS] Learning Deep Features for One-Class Classification (AnomalyDetection). Corresp...

06 Jan 2019 127
tensorlayer-tricks

How to use TensorLayer

15 Feb 2017 345
modelsummary

All Model summary in PyTorch similar to `model.summary()` in Keras

04 Mar 2019 86
torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

12 Aug 2019 568
Keras-inference-time-optimizer

Optimize layers structure of Keras model to reduce computation time

25 May 2018 157
torchview

torchview: visualize pytorch models

08 Nov 2022 803
torchtools

A High-Level training API on top of PyTorch

06 Sep 2017 17
pytorch2keras

PyTorch to Keras model convertor

16 Nov 2017 851
onnx2tf

Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The pur...

27 Sep 2022 559