Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,716Nov 22, 2022Updated 3 years ago
Alternatives and similar repositories for pytorch-playground
Users that are interested in pytorch-playground are comparing it to the libraries listed below
Sorting:
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- 95.47% on CIFAR10 with PyTorch☆6,365Feb 24, 2023Updated 3 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆23,793Sep 1, 2025Updated 6 months ago
- Classification with PyTorch.☆1,743Jun 18, 2024Updated last year
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,754Jul 15, 2024Updated last year
- A memory-efficient implementation of DenseNets☆1,539Jun 1, 2023Updated 2 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,466Jul 11, 2018Updated 7 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,819Sep 5, 2019Updated 6 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated last month
- 3.8% and 18.3% on CIFAR-10 and CIFAR-100☆1,310Aug 20, 2019Updated 6 years ago
- ☆213Nov 23, 2018Updated 7 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,232Dec 29, 2021Updated 4 years ago
- PyTorch Tutorial for Deep Learning Researchers☆32,220Aug 15, 2023Updated 2 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,818Nov 12, 2020Updated 5 years ago
- A faster pytorch implementation of faster r-cnn☆7,863May 20, 2022Updated 3 years ago
- PyTorch and Tensorflow functional model definitions☆588Feb 1, 2018Updated 8 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,439Dec 12, 2021Updated 4 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- YOLOv2 in PyTorch☆1,564Sep 29, 2021Updated 4 years ago
- Deformable Convolutional Networks☆4,116Sep 27, 2021Updated 4 years ago
- Datasets, Transforms and Models specific to Computer Vision☆17,566Updated this week
- micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantiz…☆2,271May 6, 2025Updated 10 months ago
- Collection of generative models in Pytorch version.☆2,642Apr 12, 2020Updated 5 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,386Feb 16, 2023Updated 3 years ago
- Pytorch implementation of convolutional neural network visualization techniques☆8,199Jan 1, 2025Updated last year
- DeepLab resnet v2 model in pytorch☆603Sep 5, 2023Updated 2 years ago
- CondenseNet: Light weighted CNN for mobile devices☆691Nov 11, 2019Updated 6 years ago
- PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference☆887Jul 12, 2019Updated 6 years ago
- Channel Pruning for Accelerating Very Deep Neural Networks (ICCV'17)☆1,088May 2, 2024Updated last year
- ConvNet training using pytorch☆348Feb 4, 2021Updated 5 years ago
- Semantic Segmentation Architectures Implemented in PyTorch☆3,410Oct 11, 2023Updated 2 years ago
- A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility☆6,294Aug 6, 2023Updated 2 years ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,516Jun 7, 2020Updated 5 years ago
- Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).☆4,855Jan 9, 2024Updated 2 years ago
- ☆3,247Dec 26, 2018Updated 7 years ago
- A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy☆10,243Mar 7, 2026Updated last week
- A lightweight library for PyTorch training tools and utilities☆1,720Mar 6, 2026Updated last week
- Image-to-Image Translation in PyTorch☆25,011Aug 6, 2025Updated 7 months ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,726Jul 6, 2023Updated 2 years ago