DenseNet on STL10

This repository contains PyTorch code to train a DenseNet on the STL10 dataset.

Train something fast. DenseNet, ResNet, just not VGG.

— My professor, as he caresses his GPUs

Usage

python train_stl10.py