Beispiel #1
0
 def test_vpg_cuda(self):
     validate_agent(vpg(device=CUDA),
                    AtariEnvironment("Breakout", device=CUDA))
Beispiel #2
0
 def test_vpg(self):
     validate_agent(vpg(device=CPU), AtariEnvironment("Breakout",
                                                      device=CPU))