Exemplos de main em Python

Exemplo n.º 1

Arquivo: train_test.py Projeto: zzing0907/tensorflow

 def testTrain(self):
     train.FLAGS = self._getDefaultFlags()
     train.main('')
     self.assertTrue(
         gfile.Exists(
             os.path.join(train.FLAGS.train_dir,
                          train.FLAGS.model_architecture + '.pbtxt')))
     self.assertTrue(
         gfile.Exists(
             os.path.join(train.FLAGS.train_dir,
                          train.FLAGS.model_architecture + '_labels.txt')))
     self.assertTrue(
         gfile.Exists(
             os.path.join(train.FLAGS.train_dir,
                          train.FLAGS.model_architecture + '.ckpt-1.meta')))

Exemplo n.º 2

Arquivo: train_test.py Projeto: Albert-Z-Guo/tensorflow

 def testTrain(self):
   train.FLAGS = self._getDefaultFlags()
   train.main('')
   self.assertTrue(
       gfile.Exists(
           os.path.join(train.FLAGS.train_dir,
                        train.FLAGS.model_architecture + '.pbtxt')))
   self.assertTrue(
       gfile.Exists(
           os.path.join(train.FLAGS.train_dir,
                        train.FLAGS.model_architecture + '_labels.txt')))
   self.assertTrue(
       gfile.Exists(
           os.path.join(train.FLAGS.train_dir,
                        train.FLAGS.model_architecture + '.ckpt-1.meta')))