upload via upload_folder 2025-08-03T14:02:29.798742+00:00

README.md

@@ -25,13 +25,26 @@ model-index:
       verified: false
 ---
 
-    # **REINFORCE** Agent playing **CartPole-v1**
-    This is a trained model of a **REINFORCE** agent playing **CartPole-v1**, with a similar A3C parellel update. 
+# **Vanilla-Reinforce** Agent playing **CartPole-v1**
+This is a trained model of a **Vanilla-Reinforce** agent playing **CartPole-v1**.
 
-    ## Usage
+## Usage
+### create the conda env in https://github.com/GeneHit/drl_practice
+```bash
+conda create -n drl python=3.10
+conda activate drl
+python -m pip install -r requirements.txt
+```
 
-    model = load_from_hub(repo_id="winkin119/Reinforce-CartPole", filename="model.pth")
+### play with full model
+```python
+# load the full model
+model = load_from_hub(repo_id="winkin119/Reinforce-CartPole", filename="full_model.pt")
 
-    env = gym.make("CartPole-v1")
-    ...
-    
+# Create the environment. 
+env = gym.make("CartPole-v1")
+state, _ = env.reset()
+action = model.action(state)
+...
+```
+There is also a state dict version of the model, you can check the corresponding chapter in the repo.

full_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:beef03e0babaaf209d93ab0b58ce7320774fd7e66fa83e646ea59300fa669a31
+size 74745

state_dict.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ace282d036de7ee28a0e0333af059a3bc48b65ddd9ca3b9b5fe12e259a2350d
+size 72825