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
- Code for "Scale-Equivariant Steerable Networks"☆70Updated 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
- ☆68Updated 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
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆221Updated 4 months 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
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆263Updated 2 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆118Updated 4 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆305Updated 3 years ago
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆143Updated last year
- Deep Isometric Learning for Visual Recognition (ICML 2020)☆143Updated 3 years ago
- Implementation of the reversible residual network in pytorch☆105Updated 3 years ago
- Pytorch implementation of Learning Rate Dropout.☆42Updated 5 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
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- Pytorch implementation of the image transformer for unconditional image generation☆118Updated last year
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆234Updated 4 years ago
- Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)☆207Updated 3 years ago
- A Disentangling Invertible Interpretation Network☆123Updated 4 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 4 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆146Updated 4 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆129Updated 3 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆305Updated 3 years ago
- ☆182Updated 2 years ago
- (CVPR 2021) DC-VAE: Dual Contradistinctive Generative Autoencoder☆40Updated 2 years ago
- Framework for creating (partially) reversible neural networks with PyTorch☆152Updated 3 years ago
- Combine multiple MNIST digits to create datasets with 100/1000 classes for few-shot learning/meta-learning☆101Updated 4 years ago