ray/doc/source/ray-overview/doc_test/ray_rllib.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

12 lines
270 B
Python
Raw Normal View History

from ray import air, tune
from ray.rllib.algorithms.ppo import PPO
tuner = tune.Tuner(
PPO,
run_config=air.RunConfig(
stop={"episode_len_mean": 20},
),
param_space={"env": "CartPole-v0", "framework": "torch", "log_level": "INFO"},
)
tuner.fit()