Skip to content

Latest commit

 

History

History
6 lines (4 loc) · 256 Bytes

README.md

File metadata and controls

6 lines (4 loc) · 256 Bytes

vgg16_for_CIFAR10_with_pytorch

build vgg16 with pytorch 0.4.0 for classification of CIFAR datasets

We use pytorch_gpu 0.4.0 for building net.

To run the code, you should configure your GPU first.Or you must remove all the ".cuda()" in the "vgg16.py".