Ejemplos de rainbow en Python

Ejemplo n.º 1

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

def main():
    agents = [
        atari.a2c(),
        atari.c51(),
        atari.dqn(),
        atari.ddqn(),
        atari.ppo(),
        atari.rainbow(),
    ]
    envs = [
        AtariEnvironment(env, device='cuda')
        for env in ['BeamRider', 'Breakout', 'Pong', 'Qbert', 'SpaceInvaders']
    ]
    SlurmExperiment(agents,
                    envs,
                    10e6,
                    sbatch_args={'partition': '1080ti-long'})

Ejemplo n.º 2

 def test_rainbow_cuda(self):
     validate_agent(
         rainbow(replay_start_size=64, device=CUDA),
         AtariEnvironment("Breakout", device=CUDA),
     )