RobinBruegger / RevTorch
Framework for creating (partially) reversible neural networks with PyTorch
☆144Updated 2 years ago
Related projects: ⓘ
- A fully invertible U-Net for memory efficiency in Pytorch.☆121Updated 2 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 3 years ago
- Implementation of the reversible residual network in pytorch☆101Updated 2 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆251Updated last year
- Official implementation of Fixed-Point GAN - ICCV 2019☆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 4 years ago
- RevGAN implementation in PyTorch. We extend the Pix2pix and CycleGAN framework by exploring approximately invertible architectures in 2D …☆81Updated 4 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆120Updated 8 months ago
- Implementation of Uformer, Attention-based Unet, in Pytorch☆92Updated 2 years ago
- Pytorch implementation of Learning Rate Dropout.☆42Updated 4 years ago
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆48Updated 3 years ago
- ☆164Updated 5 years ago
- Pytorch implementation of the image transformer for unconditional image generation☆114Updated last month
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆119Updated 3 years ago
- Pytorch implementations of generative models: VQVAE2, AIR, DRAW, InfoGAN, DCGAN, SSVAE☆90Updated 3 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆252Updated 3 years ago
- Batch Renormalization in Pytorch☆44Updated last year
- This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".☆39Updated 11 months ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆146Updated 8 months ago
- ☆92Updated 3 years ago
- ☆57Updated this week
- [WACV 2022] "Sandwich Batch Normalization: A Drop-In Replacement for Feature Distribution Heterogeneity" by Xinyu Gong, Wuyang Chen, Tian…☆49Updated 2 years ago
- ☆36Updated 3 months ago
- Neural Ordinary Differential Equations for Semantic Segmentation of Individual Colon Glands☆105Updated 4 years ago
- Differentiable Data Augmentation Library☆119Updated 2 years ago
- ☆50Updated 3 years ago
- [ICLR 2022] Finding an Unsupervised Image Segmenter in each of your Deep Generative Models☆76Updated 2 years ago
- MoEx (Moment Exchange)☆143Updated 3 years ago
- Official implementation for our paper "Is normalization indispensable for training deep neural networks?"☆34Updated 2 years ago
- Implementation of Evaluation Metrics for Pytorch☆35Updated 3 years ago