mustafataha5's picture
Update README.md
612c3a0 verified
# 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](https://panda-gym.readthedocs.io/en/latest/environments/pickandplace.html) environment from [Panda-Gym](https://github.com/qgallouedec/panda-gym).
The training was done using [Stable-Baselines3](https://stable-baselines3.readthedocs.io/) and uploaded to the Hugging Face Hub.
---
## πŸ“– Model Details
- **Algorithm:** SAC (Soft Actor-Critic) + HER
- **Environment:** `PandaPickAndPlace-v3`
- **Training Steps:** 800k
- **Library:** [Stable-Baselines3](https://stable-baselines3.readthedocs.io/)
- **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.
```python
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 checkpoint
- `README.md` β†’ This file
---
## πŸ™Œ Acknowledgements
- [Stable-Baselines3](https://stable-baselines3.readthedocs.io/)
- [Panda-Gym](https://github.com/qgallouedec/panda-gym)
- [Hugging Face Hub](https://huggingface.co/)
---
## πŸ“ Maintainer
Mustafa Taha
---
⚑ **Steps to use:**
1. Copy this into a file called `README.md`.
2. Place it in your Hugging Face repo (it will replace the default template).
3. 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.