aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,665Updated 2 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:
- Classification with PyTorch.☆1,726Updated 10 months ago
- A memory-efficient implementation of DenseNets☆1,533Updated last year
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,082Updated 3 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,455Updated 6 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,403Updated 3 years ago
- 95.47% on CIFAR10 with PyTorch☆6,179Updated 2 years ago
- A PyTorch implementation of DenseNet.☆839Updated 6 years ago
- PyTorch implementation of SENet☆2,320Updated 4 years ago
- Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts☆2,328Updated 2 years ago
- Deformable Convolutional Networks☆4,077Updated 3 years ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,770Updated 2 years ago
- YOLOv2 in PyTorch☆1,566Updated 3 years ago
- Faster RCNN with PyTorch☆1,743Updated 3 years ago
- Squeeze-and-Excitation Networks☆3,512Updated 6 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,526Updated 10 months ago
- Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).☆4,804Updated last year
- Collection of generative models in Pytorch version.☆2,631Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,040Updated last year
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,395Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,694Updated 2 weeks ago
- Non-local Neural Networks for Video Classification☆1,979Updated 3 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,985Updated 10 months ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,819Updated 4 years ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,275Updated 10 months ago
- A simplified implemention of Faster R-CNN that replicate performance from origin paper☆4,023Updated 4 years ago
- Simple examples to introduce PyTorch☆4,787Updated 3 years ago
- This is the PyTorch implementation of VGG network trained on CIFAR10 dataset☆351Updated 6 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,935Updated 2 weeks ago
- Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"☆2,143Updated 2 years ago
- ☆3,222Updated 6 years ago