YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
SAC + HER Agent for PandaPickAndPlace-v3 π¦Ύ
This repository contains a Soft Actor-Critic (SAC) agent trained with Hindsight Experience Replay (HER) to solve the PandaPickAndPlace-v3 environment from Panda-Gym.
The training was done using Stable-Baselines3 and uploaded to the Hugging Face Hub.
π Model Details
- Algorithm: SAC (Soft Actor-Critic) + HER
- Environment:
PandaPickAndPlace-v3
- Training Steps: 800k
- Library: Stable-Baselines3
- Replay Buffer: HER with
future
strategy - Device: Trained on GPU (
cuda
)
π Evaluation Results
The agent was evaluated for 10 episodes:
Mean reward = XXX.XX Β± YYY.YY
Please replace XXX.XX and YYY.YY with your actual evaluation results.
π Usage
You can directly load this trained agent from the Hugging Face Hub and run it inside the PandaPickAndPlace-v3
environment.
import gymnasium as gym
from stable_baselines3 import SAC
from huggingface_sb3 import load_from_hub
# Download model from Hugging Face Hub
repo_id = "mustafataha5/sac-her-PandaPickAndPlace-v3-800k" # your repo
filename = "sac_her_checkpoint_800000_steps.zip" # uploaded file
# This will download the model from HF Hub
model_path = load_from_hub(repo_id, filename)
model = SAC.load(model_path)
# Create the environment
env = gym.make("PandaPickAndPlace-v3", render_mode="human")
# Run one episode
obs, _ = env.reset()
done, truncated = False, False
while not (done or truncated):
action, _ = model.predict(obs, deterministic=True)
obs, reward, done, truncated, info = env.step(action)
env.render()
env.close()
π¦ Files inside this repo
sac_her_checkpoint_800000_steps.zip
β The trained SAC + HER model checkpointREADME.md
β This file
π Acknowledgements
π Maintainer
Mustafa Taha
β‘ Steps to use:
- Copy this into a file called
README.md
. - Place it in your Hugging Face repo (it will replace the default template).
- Commit + push.
Then, when people visit your model page, theyβll see this professional README and can copy-paste the usage code to download + run your agent.
- Downloads last month
- 3
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
π
Ask for provider support