Update README.md
Browse files
README.md
CHANGED
@@ -4,7 +4,7 @@ tags:
|
|
4 |
- reinforcement-learning
|
5 |
- stable-baselines3
|
6 |
---
|
7 |
-
# "
|
8 |
|
9 |
This is a pre-trained model of a PPO agent playing LunarLander-v2 using the [stable-baselines3](https://github.com/DLR-RM/stable-baselines3) library.
|
10 |
|
@@ -28,7 +28,7 @@ from stable_baselines3.common.evaluation import evaluate_policy
|
|
28 |
# Retrieve the model from the hub
|
29 |
## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
|
30 |
## filename = name of the model zip file from the repository
|
31 |
-
checkpoint = load_from_hub(repo_id="
|
32 |
model = PPO.load(checkpoint)
|
33 |
|
34 |
# Evaluate the agent
|
|
|
4 |
- reinforcement-learning
|
5 |
- stable-baselines3
|
6 |
---
|
7 |
+
# "Beyko7/ppo-LunarLander-v2"
|
8 |
|
9 |
This is a pre-trained model of a PPO agent playing LunarLander-v2 using the [stable-baselines3](https://github.com/DLR-RM/stable-baselines3) library.
|
10 |
|
|
|
28 |
# Retrieve the model from the hub
|
29 |
## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
|
30 |
## filename = name of the model zip file from the repository
|
31 |
+
checkpoint = load_from_hub(repo_id="Beyko7/ppo-LunarLander-v2", filename="LunarLander-v2.zip")
|
32 |
model = PPO.load(checkpoint)
|
33 |
|
34 |
# Evaluate the agent
|