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 |
|