alwynmathew / gradflow-checkLinks
Check gradient flow in Pytorch
☆85Updated 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:
- Code for "Scale-Equivariant Steerable Networks"☆72Updated 5 years ago
- ☆69Updated 4 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆119Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆142Updated 3 years ago
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆265Updated 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…☆98Updated 6 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- Pytorch Custom CUDA kernel for searchsorted☆136Updated 2 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆160Updated 4 years ago
- A Disentangling Invertible Interpretation Network☆123Updated 4 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆222Updated 8 months ago
- Deep Isometric Learning for Visual Recognition (ICML 2020)☆143Updated 3 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 4 years ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆158Updated last year
- ☆106Updated 3 years ago
- Useful PyTorch functions and modules that are not implemented in PyTorch by default☆189Updated last year
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆236Updated 4 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)☆207Updated 4 years ago
- Framework for creating (partially) reversible neural networks with PyTorch☆155Updated 3 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆306Updated 3 years ago
- Official implementation of "Energy-Based Models for Deep Probabilistic Regression" (ECCV 2020) and "How to Train Your Energy-Based Model …☆105Updated 5 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆17Updated 5 years ago
- ☆79Updated 4 years ago
- ☆182Updated 2 years ago
- Code for Scalable Uncertainty for Computer Vision with Functional Variational Inference @ CVPR 2020☆68Updated 5 years ago
- ☆165Updated 7 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Official PyTorch implementation of Harmonizing Maximum Likelihood with GANs for Multimodal Conditional Generation (ICLR 2019)☆94Updated last year
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago