hiranhsw commited on
Commit
bd839fe
·
1 Parent(s): f1d3f68

updated readme.md

Browse files
Files changed (1) hide show
  1. README.md +14 -2
README.md CHANGED
@@ -24,5 +24,17 @@ model-index:
24
  This is a trained model of a **PPO** agent playing **LunarLander-v2** using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
25
 
26
  ## Usage (with Stable-baselines3)
27
- TODO: Add your code
28
-
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  This is a trained model of a **PPO** agent playing **LunarLander-v2** using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
25
 
26
  ## Usage (with Stable-baselines3)
27
+ ```
28
+ env = make_vec_env('LunarLander-v2', n_envs=24)
29
+ model = PPO(
30
+ policy = 'MlpPolicy',
31
+ env = env,
32
+ n_steps = 1024,
33
+ batch_size = 64,
34
+ n_epochs = 16,
35
+ gamma = 0.999,
36
+ gae_lambda = 0.98,
37
+ ent_coef = 0.01,
38
+ verbose=1)
39
+ model.learn(total_timesteps=500000)
40
+ ```