Instructions to use mioti-mic/gym_xarm_01 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- stable-baselines3
How to use mioti-mic/gym_xarm_01 with stable-baselines3:
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="mioti-mic/gym_xarm_01", filename="{MODEL FILENAME}.zip", ) - Notebooks
- Google Colab
- Kaggle
Add best SAC model for gym_xarm/XarmLift-v0
Browse files
README.md
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- reinforcement-learning
|
| 5 |
+
- stable-baselines3
|
| 6 |
+
- sac
|
| 7 |
+
- robotics
|
| 8 |
+
- gymnasium
|
| 9 |
+
- mujoco
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# SB3 SAC on gym_xarm/XarmLift-v0
|
| 13 |
+
|
| 14 |
+
This repository contains a Stable-Baselines3 SAC policy trained on `gym_xarm/XarmLift-v0`.
|
| 15 |
+
|
| 16 |
+
## Files
|
| 17 |
+
- `best_model.zip` / `final_model.zip`: SB3 model checkpoint (policy + critics).
|
| 18 |
+
- `vecnormalize.pkl`: VecNormalize statistics (required for correct inference).
|
| 19 |
+
- `metadata.txt`: simple metadata about the run.
|
| 20 |
+
|
| 21 |
+
## Inference (important)
|
| 22 |
+
You must load the VecNormalize stats before running inference.
|
| 23 |
+
|
| 24 |
+
Example (pseudo):
|
| 25 |
+
1) Create env (same wrappers).
|
| 26 |
+
2) Load VecNormalize with `VecNormalize.load(...)`.
|
| 27 |
+
3) Load model with `SAC.load(...)` and set env.
|