Downloads · 30 days
7
1% of all-time downloads
araffin/tqc-RocketLander-v0
tqc-RocketLander-v0 is a reinforcement learning model from araffin. 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 TQC agent playing RocketLander-v0 using the stable-baselines3 library and the RL Zoo.
Downloads · 30 days
7
1% of all-time downloads
All-time downloads
752
Public
Repo size
7.5 MB
Likes
1
Public
Click a slice to open those files.
.zip3.9 MB · 52%
From the Hugging Face model README
This is a trained model of a TQC agent playing RocketLander-v0 using the stable-baselines3 library and the RL Zoo.
The RL Zoo is a training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
RL Zoo: https://github.com/DLR-RM/rl-baselines3-zoo<br/> SB3: https://github.com/DLR-RM/stable-baselines3<br/> SB3 Contrib: https://github.com/Stable-Baselines-Team/stable-baselines3-contrib
Gym env: https://github.com/sdsubhajitdas/Rocket_Lander_Gym
# Download model and save it into the logs/ folder
python -m rl_zoo3.load_from_hub --algo tqc --env RocketLander-v0 -orga araffin -f logs/
python enjoy.py --algo tqc --env RocketLander-v0 -f logs/
python train.py --algo tqc --env RocketLander-v0 -f logs/
# Upload the model and generate video (when possible)
python -m rl_zoo3.push_to_hub --algo tqc --env RocketLander-v0 -f logs/ -orga araffin
OrderedDict([('env_wrapper',
[{'rl_zoo3.wrappers.FrameSkip': {'skip': 4}},
{'rl_zoo3.wrappers.HistoryWrapper': {'horizon': 2}}]),
('n_timesteps', 3000000.0),
('policy', 'MlpPolicy'),
('normalize', False)])