Thomas Simonini commited on
Commit ·
a9382b8
1
Parent(s): e61d111
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ from stable_baselines3.common.vec_env import VecFrameStack
|
|
| 31 |
# Retrieve the model from the hub
|
| 32 |
## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
|
| 33 |
## filename = name of the model zip file from the repository
|
| 34 |
-
checkpoint = load_from_hub(repo_id="ThomasSimonini/ppo-SpaceInvadersNoFrameskip-v4", filename="ppo-SpaceInvadersNoFrameskip-v4")
|
| 35 |
model = PPO.load(checkpoint)
|
| 36 |
```
|
| 37 |
|
|
|
|
| 31 |
# Retrieve the model from the hub
|
| 32 |
## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
|
| 33 |
## filename = name of the model zip file from the repository
|
| 34 |
+
checkpoint = load_from_hub(repo_id="ThomasSimonini/ppo-SpaceInvadersNoFrameskip-v4", filename="ppo-SpaceInvadersNoFrameskip-v4.zip")
|
| 35 |
model = PPO.load(checkpoint)
|
| 36 |
```
|
| 37 |
|