leimao / PyTorch-Eager-Mode-Quantization-TensorRT-Acceleration
TensorRT Acceleration for PyTorch Native Eager Mode Quantization Models
☆15Updated 9 months ago
Alternatives and similar repositories for PyTorch-Eager-Mode-Quantization-TensorRT-Acceleration:
Users that are interested in PyTorch-Eager-Mode-Quantization-TensorRT-Acceleration are comparing it to the libraries listed below
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆40Updated 2 months ago
- A faster implementation of OpenCV-CUDA that uses OpenCV objects, and more!☆49Updated this week
- Model compression for ONNX☆91Updated 5 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆116Updated this week
- ☆31Updated 10 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆86Updated 3 months ago
- TensorRT 2022复赛方案： 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆138Updated 2 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated 4 months ago
- Timm model explorer☆39Updated last year
- ☆203Updated 3 years ago
- EfficientViT is a new family of vision models for efficient high-resolution vision.☆24Updated last year
- Simplify Your Visual Data Ops. Find and visualize issues with your computer vision datasets such as duplicates, anomalies, data leakage, …☆68Updated last year
- PyTorch Pruning Example☆49Updated 2 years ago
- ☆154Updated last year
- MegEngine到其他框架的转换器☆69Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Nsight Systems In Docker☆20Updated last year
- A code generator from ONNX to PyTorch code☆136Updated 2 years ago
- This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…☆64Updated this week
- A tool convert TensorRT engine/plan to a fake onnx☆38Updated 2 years ago
- A set of examples around MegEngine☆31Updated last year
- Implementing DropPath/StochasticDepth in PyTorch☆16Updated 3 years ago
- Restorers provide out-of-the-box TensorFlow implementations of SoTA image and video restoration models for tasks such as low-light enhanc…☆39Updated last year
- DeltaCNN End-to-End CNN Inference of Sparse Frame Differences in Videos☆59Updated 2 years ago
- PyTorch Lightning based training of Semantic Segmentation models☆38Updated 2 years ago
- Page for the CVPR 2023 Tutorial - Efficient Neural Networks: From Algorithm Design to Practical Mobile Deployments☆12Updated last year
- Making a PyTorch model easier than ever!☆79Updated 3 years ago
- Count number of parameters / MACs / FLOPS for ONNX models.☆91Updated 6 months ago
- Export utility for unconstrained channel pruned models☆72Updated last year
- Inference of quantization aware trained networks using TensorRT☆80Updated 2 years ago