Ejemplos de train_mnist_single_machine en Python

Ejemplo n.º 1

Archivo: convnet_test.py Proyecto: uwroute/kfac

 def testTrainMnistSingleMachine(self):
   with tf.Graph().as_default():
     # Ensure model training doesn't crash.
     #
     # Ideally, we should check that accuracy increases as the model converges,
     # but there are too few parameters for the model to effectively memorize
     # the training set the way an MLP can.
     convnet.train_mnist_single_machine(
         data_dir=None, num_epochs=1, use_fake_data=True, device="/cpu:0")

Ejemplo n.º 2

Archivo: convnet_mnist_single_main.py Proyecto: uwroute/kfac

def main(unused_argv):
    convnet.train_mnist_single_machine(FLAGS.data_dir, num_epochs=200)