Python simple_gym_spec示例

示例#1

文件： rl_trainer_lib_test.py 项目： kltony/tensor2tensor

  def test_no_crash_pendulum(self):
    hparams = trainer_lib.create_hparams(
        "ppo_continuous_action_base",
        TrainTest.test_config)

    hparams.add_hparam("environment_spec", simple_gym_spec("Pendulum-v0"))
    rl_trainer_lib.train(hparams)

示例#2

文件： rl_trainer_lib_test.py 项目： yangliuy/tensor2tensor

    def test_no_crash_pendulum(self):
        hparams = trainer_lib.create_hparams("ppo_continuous_action_base",
                                             TrainTest.test_config)

        hparams.add_hparam("environment_spec",
                           rl_models.simple_gym_spec("Pendulum-v0"))
        rl_trainer_lib.train(hparams)

示例#3

文件： rl_trainer_lib_test.py 项目： anonymusNLP/EMNLP2019

    def test_no_crash_cartpole(self):
        hparams = trainer_lib.create_hparams("ppo_discrete_action_base",
                                             TrainTest.test_config)

        hparams.add_hparam("environment_spec",
                           rl_models.simple_gym_spec("CartPole-v0"))
        rl_trainer_lib.train(hparams)