Update README.md
Browse files
README.md
CHANGED
|
@@ -26,6 +26,8 @@ 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 |
from stable_baselines3 import PPO
|
| 30 |
from huggingface_sb3 import load_from_hub
|
| 31 |
|
|
@@ -39,6 +41,4 @@ checkpoint_path = load_from_hub(
|
|
| 39 |
)
|
| 40 |
|
| 41 |
|
| 42 |
-
model = PPO.load(checkpoint_path, device="cpu")
|
| 43 |
-
```
|
| 44 |
-
Test
|
|
|
|
| 26 |
using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
|
| 27 |
|
| 28 |
## Usage (with Stable-baselines3)
|
| 29 |
+
|
| 30 |
+
```
|
| 31 |
from stable_baselines3 import PPO
|
| 32 |
from huggingface_sb3 import load_from_hub
|
| 33 |
|
|
|
|
| 41 |
)
|
| 42 |
|
| 43 |
|
| 44 |
+
model = PPO.load(checkpoint_path, device="cpu")
|
|
|
|
|
|