nd
Collection
1 item • Updated
How to use ws11yrin/ppo-MultiInputPolicy-CustomSnake-v0 with stable-baselines3:
from huggingface_sb3 import load_from_hub
checkpoint = load_from_hub(
repo_id="ws11yrin/ppo-MultiInputPolicy-CustomSnake-v0",
filename="{MODEL FILENAME}.zip",
)This is a trained model of a ppo agent playing CustomSnake-v0 using the stable-baselines3 library.
TODO: Add your code
from stable_baselines3 import ...
from huggingface_sb3 import load_from_hub
...
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="ws11yrin/ppo-MultiInputPolicy-CustomSnake-v0", filename="{MODEL FILENAME}.zip", )