deeperlearner / pytorch-template
A pytorch template files generator.
☆32Updated 8 months ago
Related projects: ⓘ
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆118Updated 9 months ago
- PyTorch implementation of some learning rate schedulers for deep learning researcher.☆86Updated 2 years ago
- Deep Learning project template for PyTorch (multi-gpu training is supported)☆134Updated last year
- Check gradient flow in Pytorch☆85Updated 5 years ago
- Simple project base template for PyTorch deep Learning project. Features clean implementation of DDP training and Hydra config.☆57Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆123Updated 2 years ago
- Implementation of Linformer for Pytorch☆244Updated 8 months ago
- Implementation of Fast Transformer in Pytorch☆171Updated 3 years ago
- Gradient Reversal Layer for Domain Adaptation☆102Updated last year
- Implementation of STAM (Space Time Attention Model), a pure and simple attention model that reaches SOTA for video classification☆126Updated 3 years ago
- Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, de…☆96Updated 2 years ago
- Learning Rate Warmup in PyTorch☆387Updated 4 months ago
- Unofficial PyTorch implementation of Google's FNet: Mixing Tokens with Fourier Transforms. With checkpoints.☆68Updated 2 years ago
- Adaptive Gradient Clipping☆114Updated last year
- A minimal pytorch package implementing a gradient reversal layer.☆154Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆103Updated 3 years ago
- ☆163Updated last year
- A simple python package to solve all your problems with pythonpath, work dir, file paths, module imports and environment variables.☆130Updated last year
- Additive margin softmax loss in pytorch☆45Updated 5 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 2 years ago
- A simple cross attention that updates both the source and target in one step☆140Updated 4 months ago
- Estimate/count FLOPS for a given neural network using pytorch☆303Updated 2 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 3 years ago
- [DEPRECATED] A knowledge distillation toolkit based on PyTorch and PyTorch Lightning.☆136Updated 7 months ago
- PyTorch implementation of "data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language" from Meta AI☆166Updated last year
- Implementation of Visual Transformer for Small-size Datasets☆116Updated 2 years ago
- Recent Advances in MLP-based Models (MLP is all you need!)☆112Updated last year
- ☆433Updated last year
- Implementation for our WACV 2021 paper "Multi-Loss Weighting with Coefficient of Variations"☆49Updated 3 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆116Updated 3 years ago