Downloads · 30 days
0
0% of all-time downloads
salohiddin94/a2c-PandaReachDense-v3
a2c-PandaReachDense-v3 is a reinforcement learning model from salohiddin94. 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.
This is a trained model of a A2C agent playing PandaReachDense-v3 using the stable-baselines3 library.
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
18
Public
Repo size
201 KB
Likes
0
Public
Click a slice to open those files.
.mp4689 KB · 75%
From the Hugging Face model README
This is a trained model of a A2C agent playing PandaReachDense-v3 using the stable-baselines3 library.
TODO: Add your code
from stable_baselines3.common.vec_env import DummyVecEnv, VecNormalize
# Load the saved statistics
eval_env = DummyVecEnv([lambda: gym.make("PandaReachDense-v3")])
eval_env = VecNormalize.load("vec_normalize.pkl", eval_env)
# We need to override the render_mode
eval_env.render_mode = "rgb_array"
# do not update them at test time
eval_env.training = False
# reward normalization is not needed at test time
eval_env.norm_reward = False
# Load the agent
model = A2C.load("a2c-PandaReachDense-v3")
mean_reward, std_reward = evaluate_policy(model, eval_env)
print(f"Mean reward = {mean_reward:.2f} +/- {std_reward:.2f}")
...