Russzheng / ONNX_custom_layerLinks
Guide on how to convert custom PyTorch layers when using ONNX.
☆22Updated 7 years ago
Alternatives and similar repositories for ONNX_custom_layer
Users that are interested in ONNX_custom_layer are comparing it to the libraries listed below
Sorting:
- ☆69Updated 5 years ago
- TVMScript kernel for deformable attention☆25Updated 3 years ago
- MnasNet snapshot☆35Updated 6 years ago
- Code for Depth-wise Separable Convolutions: Performance Investigations☆19Updated 5 years ago
- Page for the CVPR 2023 Tutorial - Efficient Neural Networks: From Algorithm Design to Practical Mobile Deployments☆12Updated 2 years ago
- Faster depthwise convolutions for PyTorch☆66Updated 5 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆122Updated 6 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 4 years ago
- The Pytorch implementation of "Location-aware Upsampling for Semantic Segmentation" (LaU)☆53Updated 5 years ago
- Algorithm-hardware Co-design for Deformable Convolution☆24Updated 4 years ago
- Rectified Convolution☆45Updated 2 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 5 years ago
- Yet Another Pytorch Distributed MobileNetV2-based Networks Implementation☆64Updated 5 years ago
- You are welcomed to join us!☆51Updated 4 years ago
- ☆19Updated 4 years ago
- Programmable Neural Network Compression☆149Updated 3 years ago
- The implementation of "Shape Adaptor: A Learnable Resizing Module" [ECCV 2020].☆73Updated 4 years ago
- Pytorch implemenation of "Learning Filter Basis for Convolutional Neural Network Compression" ICCV2019☆18Updated 3 years ago
- Implementation of knapsack pruning☆27Updated 5 years ago
- Group Sparsity: The Hinge Between Filter Pruning and Decomposition for Network Compression. CVPR2020.☆65Updated 3 years ago
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2021 -- Filter Sketch for Network Pruning☆53Updated 4 years ago
- An example of C++ extension for PyTorch.☆37Updated 5 years ago
- A implement of run-length encoding for Pytorch tensor using CUDA☆14Updated 4 years ago
- ☆51Updated 6 years ago
- This is the official implementation of "DHP: Differentiable Meta Pruning via HyperNetworks".☆59Updated 3 years ago
- Count number of parameters / MACs / FLOPS for ONNX models.☆94Updated 10 months ago
- Batch Normalization Auto-fusion for PyTorch☆32Updated 5 years ago
- Training with FP16 weights in PyTorch☆79Updated 6 years ago
- Advanced optimizer with Gradient-Centralization☆21Updated 5 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 3 years ago