Downloads · 30 days
0
0% of all-time downloads
billwong1220/ppo-mlp
ppo-mlp is a reinforcement learning model from billwong1220. 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 PPO agent playing LunarLander-v2 using the stable-baselines3 library.
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
19
Public
Repo size
557 KB
Likes
0
Public
Click a slice to open those files.
.mp4161 KB · 34%
From the Hugging Face model README
This is a trained model of a PPO agent playing LunarLander-v2 using the stable-baselines3 library.
TODO: Add your code
from stable_baselines3 import ...
from huggingface_sb3 import load_from_hub
from stable_baselines3.ppo import MlpPolicy
model = PPO(
policy="MlpPolicy",
env=env,
n_steps=1024,
batch_size=64,
n_epochs=4,
gamma=0.9999,
gae_lambda=0.98,
ent_coef=0.01,
verbose=1,
)
model.learn(total_timesteps=1000000)
model_name = "ppo-LunarLander-v2"
model.save(model_name)
...