alwynmathew / gradflow-checkLinks
Check gradient flow in Pytorch
☆86Updated 6 years ago
Alternatives and similar repositories for gradflow-check
Users that are interested in gradflow-check are comparing it to the libraries listed below
Sorting:
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 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…☆97Updated 5 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆117Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆70Updated 4 years ago
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆264Updated 2 years ago
- Framework for creating (partially) reversible neural networks with PyTorch☆152Updated 2 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated 3 months 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
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆128Updated last year
- A Disentangling Invertible Interpretation Network☆123Updated 4 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆234Updated 4 years ago
- ☆67Updated 3 years ago
- Deep Isometric Learning for Visual Recognition (ICML 2020)☆143Updated 3 years ago
- ☆104Updated 2 years ago
- Pytorch implementation of the image transformer for unconditional image generation☆118Updated 11 months ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- Kervolution Library in PyTorch (CVPR 2019 Oral)☆39Updated 5 years ago
- [ICLR2022] Code for "Learning Disentangled Representation by Exploiting Pretrained Generative Models: A Contrastive Learning View"☆133Updated 3 years ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆153Updated last year
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆143Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Pytorch implementation of Generative Models as Distributions of Functions 🌿☆149Updated last year
- Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)☆207Updated 3 years ago
- ☆182Updated 2 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆305Updated 3 years ago
- [CVPR 2021 Oral] Official PyTorch implementation of Soft-IntroVAE from the paper "Soft-IntroVAE: Analyzing and Improving Introspective Va…☆200Updated 3 years ago
- Useful PyTorch functions and modules that are not implemented in PyTorch by default☆188Updated last year
- Implementation of the reversible residual network in pytorch☆105Updated 3 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255Updated 2 years ago