sebamenabar / Pytorch-IntermediateLayerGetter
Simple easy to use module to get the intermediate values/activations from chosen Pytorch submodules. Supports submodule annidation and sequential modules.
☆58Updated 4 years ago
Alternatives and similar repositories for Pytorch-IntermediateLayerGetter:
Users that are interested in Pytorch-IntermediateLayerGetter are comparing it to the libraries listed below
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆216Updated last week
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆191Updated 6 months ago
- A tiny library for stochastic dataset caching in PyTorch.☆40Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆204Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆471Updated 2 years ago
- ☆73Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- ☆163Updated 2 years ago
- Create animations for the optimization trajectory of neural nets☆145Updated 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.☆125Updated 2 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆227Updated last month
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Compare neural networks by their feature similarity☆352Updated last year
- Implementation of LogAvgExp for Pytorch☆33Updated 2 years ago
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆127Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆75Updated 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
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 9 months ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆139Updated 9 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆41Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆160Updated 9 months ago
- ☆199Updated 2 years ago
- Efficient CUDA kernels for training convolutional neural networks with PyTorch.☆39Updated 3 weeks ago