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
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- 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
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- A tiny library for stochastic dataset caching in PyTorch.☆39Updated 9 months ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆124Updated 2 years ago
- Natural Language Descriptions of Deep Visual Features, ICLR 2022☆62Updated last year
- ☆47Updated 7 months ago
- Visualizing representations with diffusion based conditional generative model.☆83Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- PixMix: Dreamlike Pictures Comprehensively Improve Safety Measures (CVPR 2022)☆100Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆157Updated 5 months ago
- Python library for argument and configuration management☆53Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated 6 months ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆336Updated 2 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- ☆47Updated last year
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆208Updated this week
- ☆39Updated 5 months ago
- Feature extraction made simple with torchextractor☆101Updated 3 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Multidimensional indexing for tensors☆112Updated last year
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 6 months ago
- Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.☆63Updated 2 years ago