thatgeeman commited on
Commit
03ec795
1 Parent(s): 6ea878a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md CHANGED
@@ -25,6 +25,26 @@ model-index:
25
  This is a trained model of a **PPO** agent playing **LunarLander-v2**
26
  using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
27
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  ## Usage (with Stable-baselines3)
29
  TODO: Add your code
30
 
 
25
  This is a trained model of a **PPO** agent playing **LunarLander-v2**
26
  using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
27
 
28
+ ## Model parameters
29
+ ```python
30
+ model = PPO(
31
+ policy = 'MlpPolicy',
32
+ env = env,
33
+ n_steps = 1024,
34
+ batch_size = 64,
35
+ n_epochs = 10,
36
+ gamma = 0.999,
37
+ gae_lambda = 0.98,
38
+ ent_coef = 0.01,
39
+ verbose=1)
40
+
41
+ ```
42
+ Trained for 10^6 steps using
43
+
44
+ ```python
45
+ steps = 1e6
46
+ model.learn(total_timesteps=int(steps))
47
+ ```
48
  ## Usage (with Stable-baselines3)
49
  TODO: Add your code
50