Python main Examples

Example #1

 def test_sv2p(self):
     FLAGS.output_dir = tf.test.get_temp_dir()
     FLAGS.loop_hparams_set = "rlmb_tiny_sv2p"
     trainer_model_based.main(None)

Example #2

 def test_basic(self):
     FLAGS.output_dir = tf.test.get_temp_dir()
     FLAGS.loop_hparams_set = "rlmb_tiny"
     FLAGS.schedule = "train"  # skip evaluation for world model training
     trainer_model_based.main(None)

Example #3

 def _test_hparams_skip_evaluation(self, hparams_set):
     FLAGS.output_dir = tf.test.get_temp_dir()
     FLAGS.loop_hparams_set = hparams_set
     FLAGS.schedule = "train"  # skip evaluation for world model training
     trainer_model_based.main(None)

Example #4

File: trainer_model_based_recurrent_test.py Project: qixiuai/tensor2tensor

 def test_basic_recurrent(self):
   FLAGS.output_dir = tf.test.get_temp_dir()
   FLAGS.loop_hparams_set = "rlmb_tiny_recurrent"
   FLAGS.schedule = "train"  # skip evaluation for world model training
   trainer_model_based.main(None)

Example #5

File: trainer_model_based_sv2p_test.py Project: qixiuai/tensor2tensor

 def test_sv2p(self):
   FLAGS.output_dir = tf.test.get_temp_dir()
   FLAGS.loop_hparams_set = "rlmb_tiny_sv2p"
   trainer_model_based.main(None)