sebamenabar / Pytorch-IntermediateLayerGetter
Simple easy to use module to get the intermediate values/activations from chosen Pytorch submodules. Supports submodule annidation and sequential modules.
☆54Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for Pytorch-IntermediateLayerGetter
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆73Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆137Updated 9 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆208Updated this week
- A database of over 1.4 billion 3x3 convolution filters extracted from hundreds of diverse CNN models with relevant meta information (CVPR…☆30Updated last year
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆157Updated 6 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆73Updated 11 months ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆119Updated 11 months ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- ☆164Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- ☆41Updated 5 months ago
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆91Updated 9 months ago
- ☆73Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆124Updated 2 years ago
- Feature extraction made simple with torchextractor☆101Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 2 years ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆336Updated 3 months ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated 6 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆105Updated last year
- NumPy arrays, ready for human consumption☆63Updated 4 months ago
- Multidimensional indexing for tensors☆113Updated last year
- ☆131Updated last year
- Visualizing representations with diffusion based conditional generative model.☆84Updated last year
- Adaptive Gradient Clipping☆117Updated 2 years ago