Skip to content

Commit

Permalink
Update getting_started.md (#84)
Browse files Browse the repository at this point in the history
  • Loading branch information
badfilms authored Dec 2, 2023
1 parent 8ecc490 commit a820287
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docs/getting_started.md
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ This algorithm works by building up a sequence of button presses that do well in

Using ["Proximal Policy Optimization"](https://arxiv.org/abs/1707.06347) by Schulman et al., you can train an agent to play many of the games, though it takes awhile and is much faster with a GPU.

This example requires installing [OpenAI Baselines](https://github.com/openai/baselines). Once installed, you can run it:
This example requires installing [Stable Baselines](https://github.com/DLR-RM/stable-baselines3). Once installed, you can run it:

```shell
python3 -m retro.examples.ppo --game Airstriker-Genesis
Expand Down
5 changes: 3 additions & 2 deletions retro/examples/trivial_random_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,10 @@ def main():
env = retro.make(game="Airstriker-Genesis")
env.reset()
while True:
obs, rew, done, info = env.step(env.action_space.sample())
action = env.action_space.sample()
observation, reward, terminated, truncated, info = env.step(action)
env.render()
if done:
if terminated or truncated:
env.reset()
env.close()

Expand Down

0 comments on commit a820287

Please sign in to comment.