mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00

Tests all snippets we have on ray.io. There were some minor issues, which I'll fix upstream. Signed-off-by: Max Pumperla <max.pumperla@googlemail.com>
8 lines
207 B
Python
8 lines
207 B
Python
from ray import tune
|
|
from ray.rllib.agents.ppo import PPOTrainer
|
|
|
|
tune.run(
|
|
PPOTrainer,
|
|
stop={"episode_len_mean": 20},
|
|
config={"env": "CartPole-v0", "framework": "torch", "log_level": "INFO"},
|
|
)
|