Skip to content

dapello/vgg-feature-gen

Repository files navigation

pytorch-vgg-cifar10

This is the PyTorch implementation of VGG network trained on CIFAR10 dataset

Requirements.

[PyTorch] (https://github.com/pytorch/pytorch)

[torchvision] (https://github.com/pytorch/vision)

Download the model

The trained VGG model. 92.4% Accuracy VGG

Evaluation

wget https://www.cs.unc.edu/~cyfu/cifar10/model_best.pth.tar
python main.py --resume=./model_best.pth.tar -e

Train with script! (16-bit precision)

./run.sh 

Using the run.sh script to generate the training log and models of different versions of VGG in 16-bit or 32-bit precision. Then use the ipython notebook plot.ipynb to view the results.

alt text

About

generate features from a variety of vgg networks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages