Views
No views yet
1from stable_baselines3 import PPO
2from huggingface_sb3 import load_from_hub
3
4# Download the model from the hub
5repo_id = "mo7hamed1x/ppo-LunarLander-v2"
6filename = "ppo-LunarLander-v2.zip"
7
8checkpoint = load_from_hub(repo_id, filename)
9model = PPO.load(checkpoint)
10
11# Enjoy the trained agent
12# env = gym.make("LunarLander-v2")