Reinforcement Learning
stable-baselines3
LunarLander-v2
deep-reinforcement-learning
Eval Results (legacy)
Instructions to use Jojo78/ppo-LunarLander-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- stable-baselines3
How to use Jojo78/ppo-LunarLander-v2 with stable-baselines3:
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="Jojo78/ppo-LunarLander-v2", filename="{MODEL FILENAME}.zip", ) - Notebooks
- Google Colab
- Kaggle
PPO Agent playing LunarLander-v2
This is a trained model of a PPO agent playing LunarLander-v2 using the stable-baselines3 library.
Usage (with Stable-baselines3)
Create the environment
env = make_vec_env("LunarLander-v2", n_envs=16)
Defining the model, we use MultiLayerPerceptron (MLPPolicy) because the input is a vector,
if we had frames as input we would use CnnPolicy
model = PPO( policy="MlpPolicy", env=env, n_steps=1024, batch_size=64, n_epochs=4, gamma=0.999, gae_lambda=0.98, ent_coef=0.01, verbose=1, )
Training the model for 3,000,000 timesteps
model.learn(total_timesteps=3000000)
Save the model
model_name = "ppo-LunarLander-v2" model.save(model_name)
- Downloads last month
- -
Evaluation results
- mean_reward on LunarLander-v2self-reported291.20 +/- 18.45