https://github.com/blueskyson/cifar10-vgg16

PyTorch VGG16 for CIFAR10.
https://github.com/blueskyson/cifar10-vgg16

cifar10 deep-learning python pytorch vgg16

Last synced: 7 months ago
JSON representation

PyTorch VGG16 for CIFAR10.

• Host: GitHub
• URL: https://github.com/blueskyson/cifar10-vgg16
• Owner: blueskyson
• Created: 2021-12-28T12:25:49.000Z (almost 4 years ago)
• Default Branch: main
• Last Pushed: 2022-12-05T06:19:23.000Z (almost 3 years ago)
• Last Synced: 2025-02-08T22:45:00.194Z (8 months ago)
• Topics: cifar10, deep-learning, python, pytorch, vgg16
• Language: Python
• Homepage:
• Size: 431 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# cifar10-vgg16

PyTorch VGG16 for CIFAR10.

To train: `python vgg16train.py`

Here are some changeable variables in `vgg16train.py`:

```python

BATCH_SIZE = 128

LEARNING_RATE = 0.01

EPOCH = 20

FILENAME = "vgg16.pth"

```

You can also download [pre-trained model](https://github.com/blueskyson/cifar10-vgg16/releases) from releases.

To execute test application: `python main.py`

## Screenshots

![](./images/1.png)

- Show random labeled images  

  ![](./images/2.png)

- Show training accuracy  

  ![](./images/3.png)

- Show test  

  ![](./images/4.png)