RobinBruegger / RevTorchLinks
Framework for creating (partially) reversible neural networks with PyTorch
☆152Updated 2 years ago
Alternatives and similar repositories for RevTorch
Users that are interested in RevTorch are comparing it to the libraries listed below
Sorting:
- A fully invertible U-Net for memory efficiency in Pytorch.☆127Updated 3 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255Updated 2 years ago
- This Python package enables the training and inference of deep learning models for very large data, such as megapixel images, using atten…☆97Updated 5 years ago
- Implementation of the reversible residual network in pytorch☆105Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆153Updated last year
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆135Updated 3 months ago
- RevGAN implementation in PyTorch. We extend the Pix2pix and CycleGAN framework by exploring approximately invertible architectures in 2D …☆84Updated 5 years ago
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆50Updated 3 years ago
- [WACV 2022] "Sandwich Batch Normalization: A Drop-In Replacement for Feature Distribution Heterogeneity" by Xinyu Gong, Wuyang Chen, Tian…☆50Updated 3 years ago
- Fourier Image Transformer (FIT) can solve relevant image analysis tasks in Fourier space.☆98Updated 3 years ago
- Implementation of Uformer, Attention-based Unet, in Pytorch☆95Updated 3 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 2 years ago
- Pytorch implementation of the image transformer for unconditional image generation☆118Updated 11 months ago
- ☆165Updated 6 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- Neural Ordinary Differential Equations for Semantic Segmentation of Individual Colon Glands☆110Updated 5 years ago
- Batch Renormalization in Pytorch☆45Updated 2 years ago
- Official implementation of Fixed-Point GAN - ICCV 2019☆100Updated 4 years ago
- Unofficial PyTorch Implementation of EvoNorm☆122Updated 3 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 3 years ago
- This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".☆41Updated last year
- Pytorch implementation of Learning Rate Dropout.☆42Updated 5 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆141Updated 4 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 4 years ago
- Check gradient flow in Pytorch☆86Updated 6 years ago
- ☆182Updated 2 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 4 years ago
- [ICLR 2022] Finding an Unsupervised Image Segmenter in each of your Deep Generative Models☆78Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆70Updated 4 years ago