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".
forked from lavendelion/vgg16_for_CIFAR10_with_pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
bxwang2016/vgg16_for_CIFAR10_with_pytorch
About
build vgg16 with pytorch 0.4.0 for classification of CIFAR datasets
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%