Downloads · 30 days
0
ShuzhengTian/Carlo-Reinforce-unit_4
Carlo-Reinforce-unit_4 is a reinforcement learning model from ShuzhengTian. 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 Reinforce agent playing {envid} . To learn to use this model and train yours check Unit 4 of the Deep Reinforcement Learning Course: https://huggingface.co/deep-rl-course/unit4/introduction
Downloads · 30 days
0
Access
Public
Updated Oct 13, 2025
Repo size
12.4 KB
Likes
0
Public
Click a slice to open those files.
.mp495.7 KB · 85%
From the Hugging Face model README
This is a trained model of a Reinforce agent playing {env_id} . To learn to use this model and train yours check Unit 4 of the Deep Reinforcement Learning Course: https://huggingface.co/deep-rl-course/unit4/introduction
def load_policy(ckpt_path, s_size, a_size, h_size, device="cpu"):
policy = Policy(s_size, a_size, h_size).to(device)
policy.load_state_dict(torch.load(ckpt_path, map_location=device))
policy.eval()
return policy
from stable_baselines3 import ...
from huggingface_sb3 import load_from_hub
from inference import load_policy()
...