frgfm / torch-scan

Related projects ⓘ

Alternatives and complementary repositories for torch-scan

• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆304Updated 2 years ago
• Fangyh09 / pytorch-receptive-field
  Compute CNN receptive field size in pytorch in one line
  ☆349Updated 6 months ago
• antoinebrl / torchextractor
  Feature extraction made simple with torchextractor
  ☆101Updated 3 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆571Updated 6 months ago
• pabloppp / pytorch-tools
  Useful PyTorch functions and modules that are not implemented in PyTorch by default
  ☆187Updated 6 months ago
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆480Updated 3 years ago
• PhoenixDL / rising
  Provides everything needed for high performance data loading and augmentation in pytorch.
  ☆316Updated 10 months ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆608Updated 3 years ago
• 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
• shelfwise / receptivefield
  Gradient based receptive field estimation for Convolutional Neural Networks
  ☆335Updated 4 years ago
• benjs / nfnets_pytorch
  Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".
  ☆159Updated 3 years ago
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆417Updated 6 months ago
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆208Updated 6 months ago
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆360Updated 6 years ago
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆345Updated 10 months ago
• yaysummeriscoming / DALI_pytorch_demo
  Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …
  ☆194Updated 4 years ago
• bonlime / pytorch-tools
  Tool box for PyTorch
  ☆186Updated 8 months ago
• google-research / receptive_field
  Compute receptive fields of your favorite convnets
  ☆435Updated 3 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆335Updated 4 years ago
• ryul99 / pytorch-project-template
  Deep Learning project template for PyTorch (multi-gpu training is supported)
  ☆134Updated last year
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆477Updated last year
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆545Updated 3 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated last year
• microsoft / vision-longformer
  ☆241Updated 2 years ago
• google-research / diffstride
  TF/Keras code for DiffStride, a pooling layer with learnable strides.
  ☆124Updated 2 years ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆406Updated last month
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• richardkxu / distributed-pytorch
  Distributed, mixed-precision training with PyTorch
  ☆89Updated 4 years ago
• google-research / reassessed-imagenet
  Labels and other data for the paper "Are we done with ImageNet?"
  ☆185Updated 2 years ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆392Updated this week