pytorch-summary
by sksq96
Model summary in PyTorch similar to model.summary() in Keras
AI summary
Model visualizer
Provides a PyTorch API to display model architecture and statistics in a format similar to Keras' model.summary()
- stars
- 4K
- forks
- 416
- watching
- 37
Similar projects
Found by comparing what the projects do, not just their names.
Model runner
Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment.
C++ PyTorch inference toolkit
A repository providing tools and examples to serve PyTorch models as C++ inference applications.
Model converter
Converts PyTorch models to their Torch equivalents.
C++ extension library
Enables the creation of custom C++ extensions with CUDA support in PyTorch
PyTorch inference
Enables fast and efficient inference of PyTorch models on Windows 10 platforms using C++
Hadamard product layer
A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors.
pytorch/tnt1.7K
PyTorch toolkit
A collection of tools and utilities for building and training neural networks with PyTorch.
ResNet simulator
Reproduces ResNet-V3 with PyTorch for computer vision tasks
PyTorch utilities
A collection of utility functions to extend PyTorch functionality
Neural computer
An implementation of the Differentiable Neural Computer architecture in PyTorch
Model trainer
A tool to simplify training and deployment of PyTorch models on various devices and configurations
Message passing abstraction
Decoupled Neural Interfaces using Synthetic Gradients for PyTorch
PyTorch node
A ROS2 node template to run PyTorch models in C++
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
PyTorch C++ implementation
A C++ implementation of PyTorch tutorials
