nccao / mystore1

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

VGG19-Lenet-tensorflow

This is the tenet model and vgg19 model for cifar10, acc for lenet is about 84% and acc for vgg19 is about 86%.

My code shows how to write tfrecord, data argument, train, eval and model.

For building tfrecord

python3 tfrecord.py

For training

python3 train.py tenet

python3 train.py vgg19

For eval python3 eval.py tenet

python3 eavl.py vgg19

You also can change the config in config.py

About


Languages

Language:Python 100.0%