winkin119/TD3-PendulumV1

TD3 Agent playing Pendulum-v1

This is a trained model of a TD3 agent playing Pendulum-v1.

Usage

create the conda env in https://github.com/GeneHit/drl_practice

conda create -n drl python=3.10
conda activate drl
python -m pip install -r requirements.txt

play with full model

# load the full model
model = load_from_hub(repo_id="winkin119/TD3-PendulumV1", filename="td3_pendulum.pth")

# Create the environment. 
env = gym.make("Pendulum-v1")
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.