RobinBruegger / RevTorch
Framework for creating (partially) reversible neural networks with PyTorch
☆146Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for RevTorch
- Implementation of the reversible residual network in pytorch☆101Updated 2 years ago
- A fully invertible U-Net for memory efficiency in Pytorch.☆121Updated 2 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆252Updated last year
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 3 years ago
- This Python package enables the training and inference of deep learning models for very large data, such as megapixel images, using atten…☆98Updated 5 years ago
- Pytorch implementation of the image transformer for unconditional image generation☆116Updated 3 months ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆252Updated 3 years ago
- Neural Ordinary Differential Equations for Semantic Segmentation of Individual Colon Glands☆106Updated 4 years ago
- Fourier Image Transformer (FIT) can solve relevant image analysis tasks in Fourier space.☆95Updated 2 years ago
- Official implementation of Fixed-Point GAN - ICCV 2019☆98Updated 3 years ago
- RevGAN implementation in PyTorch. We extend the Pix2pix and CycleGAN framework by exploring approximately invertible architectures in 2D …☆82Updated 4 years ago
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆48Updated 3 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆122Updated 10 months ago
- ☆165Updated 5 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated 6 months ago
- Batch Renormalization in Pytorch☆45Updated last year
- Pytorch implementation of Learning Rate Dropout.☆42Updated 4 years ago
- This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".☆40Updated last year
- A PyTorch add-on for working with image mappings and displacement fields, including Spatial Transformers☆50Updated 2 years ago
- Exploiting Explainable Metrics for Augmented SGD [CVPR2022]☆45Updated 2 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆136Updated 2 years ago
- Implementation of Uformer, Attention-based Unet, in Pytorch☆93Updated 3 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 3 years ago
- Code for "Are labels necessary for neural architecture search"☆92Updated 8 months ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- ☆180Updated last year
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆150Updated 10 months ago
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆123Updated last year
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆123Updated 3 years ago
- Differentiable Data Augmentation Library☆120Updated 2 years ago