def validate(self, make_agent):
validate_agent(make_agent, GymEnvironment('Pendulum-v0'))
def test_vsarsa(self):
validate_agent(vsarsa(device=CPU, n_envs=4),
AtariEnvironment("Breakout", device=CPU))
def validate(self, make_agent):
validate_agent(make_agent, GymEnvironment('LunarLanderContinuous-v2'))
def test_vpg(self):
validate_agent(vpg(device=CPU), AtariEnvironment("Breakout",
device=CPU))
def test_vpg_cuda(self):
validate_agent(vpg(device=CUDA),
AtariEnvironment("Breakout", device=CUDA))
def test_vac_cuda(self):
validate_agent(vac(device=CUDA, n_envs=4),
AtariEnvironment("Breakout", device=CUDA))
def test_rainbow_cuda(self):
validate_agent(
rainbow(replay_start_size=64, device=CUDA),
AtariEnvironment("Breakout", device=CUDA),
)
def test_dqn(self):
validate_agent(
dqn(replay_start_size=64, device=CPU),
AtariEnvironment("Breakout", device=CPU),
)
def test_ddqn_cuda(self):
validate_agent(
ddqn(replay_start_size=64, device=CUDA),
AtariEnvironment("Breakout", device=CUDA),
)
def validate(self, make_agent):
validate_agent(make_agent, GymEnvironment('CartPole-v0'))
