Fix release 0.8.5 tests for PPO torch Breakout. (#8226)

This commit is contained in:
Sven Mika
2020-04-29 14:11:25 -07:00
committed by SangBin Cho
parent 5d2e2532e7
commit 80f443fad6
2 changed files with 2 additions and 2 deletions
@@ -52,7 +52,7 @@ atari-ppo-torch:
stop:
time_total_s: 3600
config:
use_pytorch: true,
use_pytorch: true
lambda: 0.95
kl_coeff: 0.5
clip_rewards: True
+1 -1
View File
@@ -179,7 +179,7 @@ def validate_config(config):
def get_policy_class(config):
if config.get("use_pytorch") is True:
if config["use_pytorch"]:
from ray.rllib.agents.ppo.ppo_torch_policy import PPOTorchPolicy
return PPOTorchPolicy
else: