/vgg16_for_CIFAR10_with_pytorch

build vgg16 with pytorch 0.4.0 for classification of CIFAR datasets

Primary LanguagePython

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".