vra / flopth
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
Related projects ⓘ
Alternatives and complementary repositories for flopth
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- Recent Advances in MLP-based Models (MLP is all you need!)☆112Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆207Updated last year
- Transformers w/o Attention, based fully on MLPs☆89Updated 7 months ago
- ☆164Updated last year
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆208Updated this week
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆66Updated 2 years ago
- [ICLR 2022] "Deep AutoAugment" by Yu Zheng, Zhi Zhang, Shen Yan, Mi Zhang☆62Updated 2 months ago
- A compilation of network architectures for vision and others without usage of self-attention mechanism☆77Updated last year
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆138Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆141Updated last year
- This repo is for our paper: Normalization Techniques in Training DNNs: Methodology, Analysis and Application☆84Updated 3 years ago
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs…☆181Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- ☆49Updated last year
- [ICLR 2023] "Dilated convolution with learnable spacings" Ismail Khalfaoui Hassani, Thomas Pellegrini and Timothée Masquelier☆59Updated 6 months ago
- Demystify RAM Usage in Multi-Process Data Loaders☆183Updated last year
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆73Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated last year
- An official code release of the paper RGB no more: Minimally Decoded JPEG Vision Transformers☆49Updated last year
- A simple minimal implementation of Reversible Vision Transformers☆117Updated 8 months ago
- Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, de…☆97Updated 2 years ago
- [ICLR 2022] "Unified Vision Transformer Compression" by Shixing Yu*, Tianlong Chen*, Jiayi Shen, Huan Yuan, Jianchao Tan, Sen Yang, Ji Li…☆48Updated 11 months ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆477Updated last year
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆77Updated 10 months ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆222Updated 2 years ago