Commit
·
e61d111
1
Parent(s):
3055a57
Update README.md
Browse files
README.md
CHANGED
@@ -31,8 +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="SpaceInvadersNoFrameskip-v4")
|
35 |
-
print(checkpoint)
|
36 |
model = PPO.load(checkpoint)
|
37 |
```
|
38 |
|
|
|
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 |
|