kezewang / apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆35Updated 4 years ago
Alternatives and similar repositories for apex:
Users that are interested in apex are comparing it to the libraries listed below
- PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" (CVPR 2022)☆193Updated 2 years ago
- ☆132Updated 2 years ago
- ☆39Updated 2 years ago
- (WIP) TorchUtils is a pytorch library with several useful tools and training tricks.☆87Updated 2 years ago
- ☆98Updated 3 years ago
- Official code for paper "On the Connection between Local Attention and Dynamic Depth-wise Convolution" ICLR 2022 Spotlight☆184Updated 2 years ago
- ☆192Updated 3 years ago
- Deep Interactive Thin Object Selection☆89Updated 4 years ago
- Conditional Convolutions for Instance Segmentation, achives 37.1mAP on coco val☆141Updated 4 years ago
- ☆38Updated 2 years ago
- Official MegEngine implementation of RepLKNet☆275Updated 3 years ago
- ☆108Updated 3 years ago
- A tool to convert Caffe model to Pytorch model.☆42Updated 6 years ago
- [ICCV 2021] Instances as Queries☆408Updated last year
- Warmup learning rate wrapper for Pytorch Scheduler☆41Updated 5 years ago
- Bottleneck Transformers for Visual Recognition☆278Updated 4 years ago
- [ NeurIPS2021] This is an official implementation of our paper "HRFormer: High-Resolution Transformer for Dense Prediction".☆506Updated 2 years ago
- This is an official implementation for "ResT: An Efficient Transformer for Visual Recognition".☆282Updated 2 years ago
- Spatial Transformer Networks in Pytorch.☆156Updated 4 years ago
- Object Classification Training Framework☆142Updated 3 years ago
- This is an official implementation of "Polarized Self-Attention: Towards High-quality Pixel-wise Regression"☆255Updated 3 years ago
- TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022☆397Updated 2 years ago
- PyTorch reimplementation of the paper "Involution: Inverting the Inherence of Convolution for Visual Recognition" (2D and 3D Involution) …☆105Updated 3 years ago
- Implementation of Swin Transformer with Pytorch☆111Updated 3 years ago
- HyperSeg - Official PyTorch Implementation☆213Updated 2 years ago
- Decode JPEG image on GPU using PyTorch☆90Updated last year
- Int8 StableFusion model☆41Updated 2 years ago
- ☆198Updated 9 months ago
- ☆190Updated 2 years ago
- [BMVC2020 Best Paper Award] Official Implementation for Delving deeper into anti-aliasing in convnets☆194Updated 3 years ago