Skip to content

Commit fbbfb30

Browse files
committed
port other changes from aws#1618
1 parent c968c8d commit fbbfb30

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/rl/estimator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ class RLEstimator(Framework):
6969

7070
COACH_LATEST_VERSION_TF = "0.11.1"
7171
COACH_LATEST_VERSION_MXNET = "0.11.0"
72-
RAY_LATEST_VERSION = "0.6.5"
72+
RAY_LATEST_VERSION = "0.8.5"
7373

7474
def __init__(
7575
self,

tests/data/ray_cartpole/train_ray.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,12 @@
55
from ray.tune.logger import pretty_print
66

77
# Based on https://github.com/ray-project/ray/blob/master/doc/source/rllib-training.rst#python-api
8-
ray.init(log_to_driver=False)
8+
ray.init(log_to_driver=False, webui_host="127.0.0.1")
99
config = ppo.DEFAULT_CONFIG.copy()
1010
config["num_gpus"] = int(os.environ.get("SM_NUM_GPUS", 0))
1111
checkpoint_dir = os.environ.get("SM_MODEL_DIR", "/Users/nadzeya/gym")
1212
config["num_workers"] = 1
13-
agent = ppo.PPOAgent(config=config, env="CartPole-v0")
13+
agent = ppo.PPOTrainer(config=config, env="CartPole-v0")
1414

1515
# Can optionally call agent.restore(path) to load a checkpoint.
1616

0 commit comments

Comments
 (0)