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:
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 3 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated 2 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆236Updated last week
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆208Updated 2 years ago
- A tiny library for stochastic dataset caching in PyTorch.☆43Updated last year
- Python library for argument and configuration management☆55Updated 2 years ago
- ☆75Updated 2 years ago
- Adaptive Gradient Clipping☆144Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆129Updated 3 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆284Updated 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 9 months ago
- Create animations for the optimization trajectory of neural nets☆158Updated last year
- ☆164Updated 2 years ago
- Minimal standalone example of diffusion model☆159Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆144Updated last year
- Convert scikit-learn models to PyTorch modules☆164Updated last year
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆221Updated 2 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆221Updated 5 months ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆138Updated 5 months ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.☆169Updated 5 months ago
- Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.☆68Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- ☆62Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆182Updated 4 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆435Updated 11 months ago
- Implementation of LogAvgExp for Pytorch☆36Updated 4 months ago