sooftware / pytorch-lr-scheduler
PyTorch implementation of some learning rate schedulers for deep learning researcher.
☆89Updated 2 years ago
Alternatives and similar repositories for pytorch-lr-scheduler:
Users that are interested in pytorch-lr-scheduler are comparing it to the libraries listed below
- Implementations of Recent Papers in Computer Vision☆38Updated 2 years ago
- ☆450Updated last year
- Additive margin softmax loss in pytorch☆46Updated 5 years ago
- ☆67Updated 3 years ago
- Implementation of Fast Transformer in Pytorch☆173Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- The missing distributed weighted random sampler for PyTorch☆24Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆133Updated 4 years ago
- Deep Learning project template for PyTorch (multi-gpu training is supported)☆136Updated last year
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- Implementation of Visual Transformer for Small-size Datasets☆120Updated 3 years ago
- Learning Rate Warmup in PyTorch☆404Updated 3 weeks ago
- Simple project base template for PyTorch deep Learning project. Features clean implementation of DDP training and Hydra config.☆57Updated 4 months ago
- Tool box for PyTorch☆187Updated last year
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆304Updated 3 years ago
- [DEPRECATED] A knowledge distillation toolkit based on PyTorch and PyTorch Lightning.☆139Updated last year
- Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.☆166Updated last week
- Gradient Reversal Layer for Domain Adaptation☆116Updated 2 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 2 years ago
- Pytorch cyclic cosine decay learning rate scheduler☆48Updated 3 years ago
- Implementation of Linformer for Pytorch☆276Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆275Updated 10 months ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆129Updated 4 months ago
- PyTorch samplers that output roughly balanced batches with support for multilabel datasets☆54Updated last year
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆130Updated last week
- PyTorch implementation of "Squeezeformer: An Efficient Transformer for Automatic Speech Recognition" (NeurIPS 2022)☆139Updated 2 years ago
- Localization of thoracic abnormalities model based on VinBigData (top 1%)☆43Updated 3 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆233Updated last year
- PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers☆226Updated 3 years ago
- Simple MAE (masked autoencoders) with pytorch and pytorch-lightning.☆41Updated last year