View model summaries in PyTorch!
MIT License
Model summary in PyTorch similar to `model.summary()` in Keras
Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT...
torchview: visualize pytorch models
All Model summary in PyTorch similar to `model.summary()` in Keras
PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/impleme...
Model analyzer in PyTorch
[5 FPS - 150 FPS] Learning Deep Features for One-Class Classification (AnomalyDetection). Corresp...
An unofficial styleguide and best practices summary for PyTorch
Toolbox for vision tasks. Pre-trained vision backbones on ImageNet with PyTorch Lightning 🚀
Exploration of Self-Semi-Supervised learning for handling unlabeled data
C++ library based on tensorrt integration
A JavaScript library like PyTorch, with GPU acceleration.
Flops counter for convolutional networks in pytorch framework
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xceptio...
Shape and dimension inference (Keras-like) for PyTorch layers and neural networks