sebamenabar / Pytorch-IntermediateLayerGetter
Simple easy to use module to get the intermediate values/activations from chosen Pytorch submodules. Supports submodule annidation and sequential modules.
☆59Updated 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
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆129Updated 4 months ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆125Updated 3 years ago
- ☆73Updated 2 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated last week
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 2 years ago
- Python library for argument and configuration management☆54Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆147Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- ☆164Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated last year
- Train ImageNet *fast* in 500 lines of code with FFCV☆141Updated 10 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Implementation of LogAvgExp for Pytorch☆34Updated 3 years ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne …☆115Updated 2 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆212Updated 2 years ago
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆100Updated last month
- Convert scikit-learn models to PyTorch modules☆161Updated 10 months ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- ☆22Updated 5 months ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆343Updated 7 months ago
- Adaptive Gradient Clipping☆129Updated 2 years ago
- ☆201Updated 2 years ago
- ☆47Updated 2 years ago