sebamenabar / Pytorch-IntermediateLayerGetterLinks
Simple easy to use module to get the intermediate values/activations from chosen Pytorch submodules. Supports submodule annidation and sequential modules.
☆60Updated 5 years ago
Alternatives and similar repositories for Pytorch-IntermediateLayerGetter
Users that are interested in Pytorch-IntermediateLayerGetter are comparing it to the libraries listed below
Sorting:
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated last year
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 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.☆129Updated 6 months ago
- ☆163Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆155Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆74Updated 2 years ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆349Updated last month
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆163Updated last year
- A tiny library for stochastic dataset caching in PyTorch.☆42Updated last year
- ☆15Updated 2 years ago
- Adaptive Gradient Clipping☆133Updated 2 years ago
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated 2 months ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 2 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated last year
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆139Updated 4 years ago
- PixMix: Dreamlike Pictures Comprehensively Improve Safety Measures (CVPR 2022)☆107Updated 2 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆42Updated 3 years ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆198Updated 9 months ago
- ☆61Updated last year
- Visualizing representations with diffusion based conditional generative model.☆95Updated 2 years ago