Ejemplos de a2c en Python

Ejemplo n.º 1

Archivo: parallel_env_experiment_test.py Proyecto: sadiakhaf/autonomous-learning-library

 def test_writes_loss(self):
     experiment = MockExperiment(a2c(),
                                 self.env,
                                 quiet=True,
                                 write_loss=True)
     self.assertTrue(experiment._writer.write_loss)
     experiment = MockExperiment(a2c(),
                                 self.env,
                                 quiet=True,
                                 write_loss=False)
     self.assertFalse(experiment._writer.write_loss)

Ejemplo n.º 2

Archivo: parallel_env_experiment_test.py Proyecto: sadiakhaf/autonomous-learning-library

 def setUp(self):
     np.random.seed(0)
     torch.manual_seed(0)
     self.env = GymEnvironment('CartPole-v0')
     self.experiment = MockExperiment(a2c(), self.env, quiet=True)
     for i, env in enumerate(self.experiment._envs):
         env.seed(i)

Ejemplo n.º 3

Archivo: experiment_test.py Proyecto: eco-energy/autonomous-learning-library

 def test_runs_multi_env(self):
     experiment = MockExperiment(a2c(n_envs=3),
                                 self.env,
                                 quiet=True,
                                 episodes=3)
     self.assertEqual(
         len(experiment._writer.data["evaluation/returns/episode"]
             ["values"]), 3)

Ejemplo n.º 4

Archivo: experiment.py Proyecto: vwxyzjn/autonomous-learning-library

def main():
    device = 'cpu'
    timesteps = 40000
    run_experiment(
        [dqn(), a2c()],
        [
            GymEnvironment('CartPole-v0', device),
            GymEnvironment('Acrobot-v1', device)
        ],
        timesteps,
    )
    plot_returns_100('runs', timesteps=timesteps)

Ejemplo n.º 5

Archivo: test.py Proyecto: eco-energy/autonomous-learning-library

 def test_a2c_(self):
     self.validate(a2c())