File tree 2 files changed +3
-3
lines changed
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -69,7 +69,7 @@ class RLEstimator(Framework):
69
69
70
70
COACH_LATEST_VERSION_TF = "0.11.1"
71
71
COACH_LATEST_VERSION_MXNET = "0.11.0"
72
- RAY_LATEST_VERSION = "0.6 .5"
72
+ RAY_LATEST_VERSION = "0.8 .5"
73
73
74
74
def __init__ (
75
75
self ,
Original file line number Diff line number Diff line change 5
5
from ray .tune .logger import pretty_print
6
6
7
7
# 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" )
9
9
config = ppo .DEFAULT_CONFIG .copy ()
10
10
config ["num_gpus" ] = int (os .environ .get ("SM_NUM_GPUS" , 0 ))
11
11
checkpoint_dir = os .environ .get ("SM_MODEL_DIR" , "/Users/nadzeya/gym" )
12
12
config ["num_workers" ] = 1
13
- agent = ppo .PPOAgent (config = config , env = "CartPole-v0" )
13
+ agent = ppo .PPOTrainer (config = config , env = "CartPole-v0" )
14
14
15
15
# Can optionally call agent.restore(path) to load a checkpoint.
16
16
You can’t perform that action at this time.
0 commit comments