Downloads · 30 days
0
0% of all-time downloads
hhmm1122/fetch-reach-sac-her
fetch-reach-sac-her is a reinforcement learning model from hhmm1122. 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.
A Soft Actor-Critic (SAC) agent with Hindsight Experience Replay (HER), trained to solve the MuJoCo FetchReach task: move the gripper to a target position from sparse reward. Used here as a fast pipeline-validation ba…
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
49
Public
Repo size
6.1 MB
Likes
0
Public
Click a slice to open those files.
.zip6.1 MB · 100%
From the Hugging Face model README
A Soft Actor-Critic (SAC) agent with Hindsight Experience Replay (HER), trained to solve the MuJoCo FetchReach task: move the gripper to a target position from sparse reward. Used here as a fast pipeline-validation baseline before the harder PickAndPlace task.
import gymnasium as gym, gymnasium_robotics
from stable_baselines3 import SAC
from stable_baselines3.common.buffers import DictReplayBuffer
from huggingface_hub import hf_hub_download
gym.register_envs(gymnasium_robotics)
path = hf_hub_download("hhmm1122/fetch-reach-sac-her", "best_model.zip")
env = gym.make("FetchReach-v4", max_episode_steps=50)
model = SAC.load(path, env=env, custom_objects={
"replay_buffer_class": DictReplayBuffer, "replay_buffer_kwargs": {}, "buffer_size": 1})
n_sampled_goal=4, goal_selection_strategy="future")Full code: https://github.com/IAmHassanMehmood/rl-fetch-manipulation1~--- library_name: stable-baselines3 tags:
A Soft Actor-Critic (SAC) agent with Hindsight Experience Replay (HER), trained to solve the MuJoCo FetchReach task: move the gripper to a target position from sparse reward. Used here as a fast pipeline-validation baseline before the harder PickAndPlace task.
import gymnasium as gym, gymnasium_robotics
from stable_baselines3 import SAC
from stable_baselines3.common.buffers import DictReplayBuffer
from huggingface_hub import hf_hub_download
gym.register_envs(gymnasium_robotics)
path = hf_hub_download("hhmm1122/fetch-reach-sac-her", "best_model.zip")
env = gym.make("FetchReach-v4", max_episode_steps=50)
model = SAC.load(path, env=env, custom_objects={
"replay_buffer_class": DictReplayBuffer, "replay_buffer_kwargs": {}, "buffer_size": 1})
n_sampled_goal=4, goal_selection_strategy="future")Full code: https://github.com/IAmHassanMehmood/rl-fetch-manipulation