mirror of
https://github.com/vale981/ray
synced 2025-03-10 05:16:49 -04:00
9 lines
207 B
Python
9 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"},
|
||
|
|
)
|