Views
No views yet

Pusher-v5 (7-DOF MuJoCo Robot Arm)
pip install gymnasium[mujoco] stable-baselines3[extra]1import gymnasium as gym
2from stable_baselines3 import PPO
3
4# Load environment
5env = gym.make("Pusher-v5", render_mode="human")
6
7# Load model from local zip
8model = PPO.load("ppo_pusher.zip")
9
10# Evaluate
11obs, _ = env.reset()
12for _ in range(200):
13 action, _ = model.predict(obs, deterministic=True)
14 obs, reward, terminated, truncated, _ = env.step(action)
15 if terminated or truncated:
16 obs, _ = env.reset()
17env.close()