Python train_mnist_multitower Beispiele

Beispiel #1

 def testTrainMnistMultitower(self):
     with tf.Graph().as_default():
         # Ensure model training doesn't crash.
         convnet.train_mnist_multitower(data_dir=None,
                                        num_epochs=1,
                                        num_towers=2,
                                        use_fake_data=True)

Beispiel #2

def main(argv):
  _ = argv

  if FLAGS.num_towers > 1:
    convnet.train_mnist_multitower(
        FLAGS.data_dir, num_epochs=200, num_towers=FLAGS.num_towers)
  else:
    convnet.train_mnist_single_machine(FLAGS.data_dir, num_epochs=200)

Beispiel #3

Datei: convnet_mnist_multi_tower_main.py Projekt: HaoREN211/python_machine-learning

def main(unused_argv):
  _ = unused_argv
  assert FLAGS.num_towers > 1
  devices = ["/gpu:{}".format(tower_id) for tower_id in range(FLAGS.num_towers)]
  convnet.train_mnist_multitower(
      FLAGS.data_dir,
      num_epochs=200,
      num_towers=FLAGS.num_towers,
      devices=devices)

Beispiel #4

Datei: convnet_test.py Projekt: BhaskarNallani/tensorflow

 def testTrainMnistMultitower(self):
   with tf.Graph().as_default():
     # Ensure model training doesn't crash.
     convnet.train_mnist_multitower(
         data_dir=None, num_epochs=1, num_towers=2, use_fake_data=True)