Find file History
cheerss update the calculation of num_batches_per_epoch
the number of batches per epoch also depends on the number of gpus. it should be `num_batches_per_epoch = (cifar10.NUM_EXAMPLES_PER_EPOCH_FOR_TRAIN / FLAGS.batch_size / FLAGS.num_gpus)`
Latest commit c803290 Nov 1, 2018

NOTE: For users interested in multi-GPU, we recommend looking at the newer cifar10_estimator example instead.

CIFAR-10 is a common benchmark in machine learning for image recognition.

Code in this directory demonstrates how to use TensorFlow to train and evaluate a convolutional neural network (CNN) on both CPU and GPU. We also demonstrate how to train a CNN over multiple GPUs.

Detailed instructions on how to get started available at: