def test_sv2p(self): FLAGS.output_dir = tf.test.get_temp_dir() FLAGS.loop_hparams_set = "rlmb_tiny_sv2p" trainer_model_based.main(None)
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)
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)
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)
def test_sv2p(self): FLAGS.output_dir = tf.test.get_temp_dir() FLAGS.loop_hparams_set = "rlmb_tiny_sv2p" trainer_model_based.main(None)