← all repositories

aaron-xichen/pytorch-playground

PyTorch playground providing pretrained CNN models, standard dataset loaders, training scripts, and model quantization utilities.

pytorch-playground
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository offers a collection of predefined deep learning models and dataset fetchers for PyTorch beginners. It includes popular architectures like ResNet, VGG, Inception, and SqueezeNet along with standard benchmarks (MNIST, CIFAR-10/100, SVHN, STL-10). The project also provides training scripts and a quantization module supporting linear, minmax, and non-linear methods to compress models to specified bit-widths.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.