Skip to content

Latest commit

 

History

History
23 lines (12 loc) · 441 Bytes

README.md

File metadata and controls

23 lines (12 loc) · 441 Bytes

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