mcimpoi / tf-slim-mnist

MNIST tutorial with Tensorflow Slim (tf-slim)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tf-slim-mnist

MNIST tutorial with Tensorflow Slim (tf-slim)

Setting up data

run convert_to_records.py to create [train, validation, test].tfrecords files containing MNIST data by default (unless you specify --directory) they will be put into /tmp/data

Running

Run the training, validation, and tensorboard concurrently. The results of the training and validation should show up in tensorboard.

Running the training

run mnist_train.py which will read train.tfrecords using an input queue and output its model checkpoints, and summaries to the log directory (you can specify it with --log_dir)

Running the validation

run mnist_eval.py which will read validation.tfrecords using an input queue, and also read the train models checkpoints from log/train (by default). It will then load the model at that checkpoint and run it on the validation examples, outputting the summaries and log to its own folder log/eval (you can specify it with --log_dir)

Running tensorboard

Tensorboard allows you to keep track of your training in a nice and visual way. It will read the logs from the training and validation and should update on its own though you may have to refresh the page manually sometimes.

Make sure both training and validation output their summaries to one log directory and preferably under their own folder. Run tensorboard --logdir=log (replace log with your own log folder if you changed it).

If each process has its own folder then train and validation should have their own colour and checkbox

About

MNIST tutorial with Tensorflow Slim (tf-slim)

License:Apache License 2.0


Languages

Language:Python 100.0%