Views
No views yet
PandaPickAndPlace-v3future strategycuda)PandaPickAndPlace-v3 environment.1import gymnasium as gym
2from stable_baselines3 import SAC
3from huggingface_sb3 import load_from_hub
4
5# Download model from Hugging Face Hub
6repo_id = "mustafataha5/sac-her-PandaPickAndPlace-v3-800k" # your repo
7filename = "sac_her_checkpoint_800000_steps.zip" # uploaded file
8
9# This will download the model from HF Hub
10model_path = load_from_hub(repo_id, filename)
11model = SAC.load(model_path)
12
13# Create the environment
14env = gym.make("PandaPickAndPlace-v3", render_mode="human")
15
16# Run one episode
17obs, _ = env.reset()
18done, truncated = False, False
19
20while not (done or truncated):
21 action, _ = model.predict(obs, deterministic=True)
22 obs, reward, done, truncated, info = env.step(action)
23 env.render()
24
25env.close()sac_her_checkpoint_800000_steps.zip → The trained SAC + HER model checkpointREADME.md → This fileREADME.md.