pytorch-summary
Model visualizer
Provides a PyTorch API to display model architecture and statistics in a format similar to Keras' model.summary()
Model summary in PyTorch similar to model.summary() in Keras
4k stars
37 watching
416 forks
Language: Python
last commit: 12 months ago deep-learningkeraspytorchsummary
Related projects:
| Repository | Description | Stars |
|---|---|---|
 slabban/ros2_pytorch_cuda | Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment. | 7 |
 wizaron/pytorch-cpp-inference | A repository providing tools and examples to serve PyTorch models as C++ inference applications. | 226 |
 clcarwin/convert_torch_to_pytorch | Converts PyTorch models to their Torch equivalents. | 541 |
 pytorch/extension-cpp | Enables the creation of custom C++ extensions with CUDA support in PyTorch | 1,031 |
 zccyman/pytorch-inference | Enables fast and efficient inference of PyTorch models on Windows 10 platforms using C++ | 89 |
 sniklaus/pytorch-extension | A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. | 117 |
| pytorch/tnt | A collection of tools and utilities for building and training neural networks with PyTorch. | 1,669 |
 prlz77/resnext.pytorch | Reproduces ResNet-V3 with PyTorch for computer vision tasks | 511 |
 mrdrozdov-github/pytorch-extras | A collection of utility functions to extend PyTorch functionality | 50 |
 ypxie/pytorch-neucom | An implementation of the Differentiable Neural Computer architecture in PyTorch | 94 |
 huggingface/accelerate | A tool to simplify training and deployment of PyTorch models on various devices and configurations | 8,056 |
 koz4k/dni-pytorch | Decoupled Neural Interfaces using Synthetic Gradients for PyTorch | 236 |
 klintan/ros2_pytorch | A ROS2 node template to run PyTorch models in C++ | 39 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,236 |
 prabhuomkar/pytorch-cpp | A C++ implementation of PyTorch tutorials | 1,978 |