Downloads · 30 days
3
15% of all-time downloads
Junfeel/pusher-v5-pro
pusher-v5-pro is a reinforcement learning model from Junfeel. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. It is set up for stable-baselines3.
This model is a high-performance PPO agent controlling a 7-DOF robotic arm to push a cylinder towards a target in Gymnasium MuJoCo Pusher-v5.
Downloads · 30 days
3
15% of all-time downloads
All-time downloads
20
Public
Repo size
176 KB
Likes
0
Public
Click a slice to open those files.
.zip176 KB · 99%
From the Hugging Face model README
Pusher-v5This model is a high-performance PPO agent controlling a 7-DOF robotic arm to push a cylinder towards a target in Gymnasium MuJoCo Pusher-v5.
Trained and exported directly via MuJoCo Pusher RL Studio Pro.
Pusher-v5 (Gymnasium / MuJoCo)import gymnasium as gym
from stable_baselines3 import PPO
env = gym.make("Pusher-v5", render_mode="human")
model = PPO.load("ppo_pusher_latest.zip")
obs, _ = env.reset()
for _ in range(1000):
action, _ = model.predict(obs, deterministic=True)
obs, reward, done, truncated, _ = env.step(action)
if done or truncated:
obs, _ = env.reset()
env.close()