You can deploy this model and start earning money today!

ShuzhengTian/
Carlo-Reinforce-unit_4

Name: Carlo-Reinforce-unit_4
Availability: InStock

stable-baselines3

CartPole-v1

deep-reinforcement-learning

reinforcement-learning

model-index

Views

No views yet

Reinforce Agent playing {CartPole-v1}

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

Usage (with Stable-baselines3)

python

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()

...

ShuzhengTian/Carlo-Reinforce-unit_4

Reinforce Agent playing {CartPole-v1}

Usage (with Stable-baselines3)

ShuzhengTian/
Carlo-Reinforce-unit_4