Downloads · 30 days
9
43% of all-time downloads
humnrdble/DeepRL-unit1-optuna
DeepRL-unit1-optuna is a reinforcement learning model from humnrdble. 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
9
43% of all-time downloads
All-time downloads
21
Public
Repo size
281 KB
Likes
0
Public
Click a slice to open those files.
.ipynb831 KB · 64%
From the Hugging Face model README
This is a trained model of a PPO agent playing LunarLander-v2 using the stable-baselines3 library.
Made as part of the Deep RL course: https://huggingface.co/learn/deep-rl-course. Tuned with Optuna, as introduced in the course. This is my first successful attempt of using Optuna, so do not expect the code or parameters to be ideal!
I was able to improve upon my result from Unit1, https://huggingface.co/humnrdble/DeepRL-unit1. Both models were trained for 1500000 steps. The video of my first attempt certainly looks smoother, but scores worse.
The code is available in unit1-notebook-tuned.ipynb, but no attempt was made to make it particularly legible.
Hyperparameters deviating from the Stable-baselines3 baseline:
TODO: Add your code
from stable_baselines3 import ...
from huggingface_sb3 import load_from_hub
...