def test_vpg_cuda(self): validate_agent(vpg(device=CUDA), AtariEnvironment("Breakout", device=CUDA))
def test_vpg(self): validate_agent(vpg(device=CPU), AtariEnvironment("Breakout", device=CPU))