frgfm / torch-scanView external linksLinks
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
☆224Mar 24, 2025Updated 10 months ago
Alternatives and similar repositories for torch-scan
Users that are interested in torch-scan are comparing it to the libraries listed below
Sorting:
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆328Dec 15, 2025Updated last month
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 6 months ago
- Official PyTorch code for U-Noise: Learnable Noise Masks for Interpretable Image Segmentation (ICIP 2021)☆42Oct 21, 2021Updated 4 years ago
- Code for the ICML 2021 and ICLR 2022 papers: Skew Orthogonal Convolutions, Improved deterministic l2 robustness on CIFAR-10 and CIFAR-100☆18Feb 20, 2022Updated 3 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- 🛠 Toolbox to extend PyTorch functionalities☆421May 13, 2024Updated last year
- Rectified Convolution☆45Oct 16, 2022Updated 3 years ago
- Batch normalization fusion for PyTorch. This is an archived repository, which is not maintained.☆196Apr 6, 2020Updated 5 years ago
- Transformer for low-level vision applications, such as image restoration (denosing, super resolution, deblur)☆54May 5, 2022Updated 3 years ago
- Flops counter for neural networks in pytorch framework☆2,970Aug 20, 2025Updated 5 months ago
- Model analyzer in PyTorch☆1,502Mar 19, 2023Updated 2 years ago
- PyTorch Code for the Paper: "Exploiting Uncertainty of Loss Landscape for Stochastic Optimization [Bhaskara et al. (2019)]☆15Dec 8, 2025Updated 2 months ago
- (CVPR 2021, Oral) Dynamic Slimmable Network☆231Dec 31, 2021Updated 4 years ago
- [MAI@CVPR 2021] AsymmNet: Towards ultralight convolution neural networks using asymmetrical bottlenecks.☆27May 30, 2024Updated last year
- [CVPR 2021] Unsupervised Degradation Representation Learning for Blind Super-Resolution☆415Jun 17, 2022Updated 3 years ago
- Code release for Deep Incubation (https://arxiv.org/abs/2212.04129)☆91Mar 16, 2023Updated 2 years ago
- Learning Features with Parameter-Free Layers, ICLR 2022☆84May 3, 2023Updated 2 years ago
- ☆18Jul 24, 2023Updated 2 years ago
- [ECCV 2020] Gradient-Induced Co-Saliency Detection☆40Aug 18, 2020Updated 5 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,989Jun 16, 2024Updated last year
- Simple tutorials on Pytorch DDP training☆286Aug 19, 2022Updated 3 years ago
- Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-…☆2,285Dec 15, 2025Updated last month
- Lightweight Transformer for Multi-modal Tasks☆16Dec 9, 2022Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Jun 14, 2023Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,161Mar 22, 2024Updated last year
- [ICLR 2021 Spotlight] "CPT: Efficient Deep Neural Network Training via Cyclic Precision" by Yonggan Fu, Han Guo, Meng Li, Xin Yang, Yinin…☆31Mar 2, 2024Updated last year
- SOIT: Segmenting Objects with Instance-Aware Transformers☆14Jun 6, 2022Updated 3 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Nov 28, 2022Updated 3 years ago
- [ECCV'20 Oral] MutualNet: Adaptive ConvNet via Mutual Learning from Network Width and Resolution☆159Oct 4, 2022Updated 3 years ago
- Visualize PyTorch tensors with a single line of code.☆704Aug 29, 2025Updated 5 months ago
- Code repository for ‘Adaptive Differential Denoising for Respiratory Sounds Classification’☆20Dec 19, 2025Updated last month
- Reinforcement learning modular with pytorch☆11Jan 18, 2021Updated 5 years ago
- View model summaries in PyTorch!☆2,901Jan 26, 2026Updated 2 weeks ago
- Learning Cascaded Convolutional Networks for Blind Single Image Super-Resolution☆21Jul 15, 2020Updated 5 years ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Jul 21, 2023Updated 2 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,224Jan 12, 2026Updated last month
- PyProf2: PyTorch Profiling tool☆82Jun 25, 2020Updated 5 years ago
- Neural Sparse Representation for Image Restoration☆31Jun 9, 2020Updated 5 years ago
- This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.☆596May 16, 2023Updated 2 years ago