Downloads · 30 days
1
13% of all-time downloads
winkin119/MBPO-ReacherV5
MBPO-ReacherV5 is a reinforcement learning model from winkin119. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product.
This is a trained model of a MBPO agent playing Reacher-v5.
Downloads · 30 days
1
13% of all-time downloads
All-time downloads
8
Public
Repo size
245 KB
Likes
0
Public
Click a slice to open those files.
.0193 KB · 74%
From the Hugging Face model README
This is a trained model of a MBPO agent playing Reacher-v5.
conda create -n drl python=3.12
conda activate drl
python -m pip install -r requirements.txt
# load the full model
model = load_from_hub(repo_id="winkin119/MBPO-ReacherV5", filename="full_model.pt")
# Create the environment.
env = gym.make("Reacher-v5")
state, _ = env.reset()
action = model.action(state)
...
There is also a state dict version of the model, you can check the corresponding definition in the repo.