Views
No views yet

1from huggingface_sb3 import load_from_hub
2from stable_baselines3.common.evaluation import evaluate_policy
3
4repo_id = "salc2/rl-lunar-lander-v3-1Msteps" # The repo_id
5filename = "models/ppo_lunar_steps1_M.zip" # The model filename.zip
6
7checkpoint = load_from_hub(repo_id, filename)
8model = PPO.load(checkpoint)
9
10eval_env = gym.make("LunarLander-v3")
11mean_reward, std_reward = evaluate_policy(model, eval_env, n_eval_episodes=10, deterministic=True)
12print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
13...