traveller59 / torch2trt
convert torch module to tensorrt network or tvm function
☆88Updated 5 years ago
Alternatives and similar repositories for torch2trt:
Users that are interested in torch2trt are comparing it to the libraries listed below
- CUDA implementation of NMS for PyTorch☆86Updated 5 years ago
- ☆138Updated 5 years ago
- ☆79Updated 6 years ago
- Benchmark of TVM quantized model on CUDA☆111Updated 4 years ago
- Parallel CUDA implementation of NON maximum Suppression☆79Updated 4 years ago
- Partial Order Pruning: for Best Speed/Accuracy Trade-off in Neural Architecture Search☆149Updated 4 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆102Updated 4 years ago
- TensorRT Net Wrapper☆95Updated 5 years ago
- ☆44Updated 6 years ago
- PSROIAlign with multi-batch training support in PyTorch☆24Updated 5 years ago
- TensorRT prelu and slice☆39Updated 6 years ago
- Light-Head R-CNN in PyTorch☆93Updated 5 years ago
- Implementations of CVPR 2019 paper Distilling Object Detectors with Fine-grained Feature Imitation☆27Updated 2 years ago
- A YOLOv3 model in caffe☆42Updated 6 years ago
- Simulate quantization and quantization aware training for MXNet-Gluon models.☆46Updated 5 years ago
- ☆37Updated 6 years ago
- This is a CNN Analyzer tool, based on Netscope by dgschwend/netscope☆41Updated 7 years ago
- Mobilenet model converted from tensorflow☆48Updated 6 years ago
- Pelee(NeurIPS'18)-TensorRT Implementation (Caffe Parser)☆86Updated 5 years ago
- TensorRT plugin forDCNv2 layer in ONNX model☆60Updated 4 years ago
- centernet☆74Updated 5 years ago
- ☆26Updated 4 years ago
- This is CNN forward implementation.☆15Updated 3 years ago
- Android demo for dabnn☆20Updated 5 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆57Updated 5 years ago
- Lightweight Networks such as MobileNet, ShuffleNet and ThunderNet implemented in Pytorch☆72Updated 4 years ago
- ☆33Updated 5 years ago
- Using TensorRT to implement and accelerate YOLO v3. Multi-scale and NMS are included. The acceleration ratio reaches 3 compared to the o…☆43Updated 6 years ago
- This repo contains code for *FD-MobileNet: Improved MobileNet with A Fast Downsampling Strategy*.☆35Updated 6 years ago
- Rethinking the Smaller-Norm-Less-Informative Assumption in Channel Pruning of Convolution Layers https://arxiv.org/abs/1802.00124☆71Updated 6 years ago