244aafdcf89ae814975c8c4e3faf0bd4995c7878,rllib/utils/exploration/tests/test_curiosity.py,TestCuriosity,test_curiosity_on_large_frozen_lake,#TestCuriosity#,88

Before Change

        config["horizon"] = 40
        config["num_workers"] = 0  // local only
        config["train_batch_size"] = 512
        config["num_sgd_iter"] = 10

        num_iterations = 30
        for _ in framework_iterator(config, frameworks="torch"):
            // W/ Curiosity.

After Change

            for _ in range(num_iterations):
                result = trainer.train()
                print(result)
                if result["episode_reward_mean"] >= 0.001:
                    print("Learnt something!")
                    learnt = True
                    break
            trainer.stop()
            self.assertTrue(learnt)

            // // W/o Curiosity. Expect to learn nothing.

Instances