ianspektor commited on
Commit
f8740b8
1 Parent(s): 049f7fa

Push Reinforce agent to the Hub

Browse files
Files changed (6) hide show
  1. .gitattributes +1 -0
  2. README.md +26 -0
  3. hyperparameters.json +1 -0
  4. model.pt +3 -0
  5. replay.mp4 +3 -0
  6. results.json +1 -0
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - CartPole-v1
4
+ - reinforce
5
+ - reinforcement-learning
6
+ - custom-implementation
7
+ - deep-rl-class
8
+ model-index:
9
+ - name: reinforce-CartPole-v1
10
+ results:
11
+ - metrics:
12
+ - type: mean_reward
13
+ value: 51.50 +/- 20.49
14
+ name: mean_reward
15
+ task:
16
+ type: reinforcement-learning
17
+ name: reinforcement-learning
18
+ dataset:
19
+ name: CartPole-v1
20
+ type: CartPole-v1
21
+ ---
22
+
23
+ # **Reinforce** Agent playing **CartPole-v1**
24
+ This is a trained model of a **Reinforce** agent playing **CartPole-v1** .
25
+ To learn to use this model and train yours check Unit 5 of the Deep Reinforcement Learning Class: https://github.com/huggingface/deep-rl-class/tree/main/unit5
26
+
hyperparameters.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"h_size": 32, "n_training_episodes": 5000, "n_evaluation_episodes": 20, "max_t": 1000, "gamma": 0.99, "lr": 0.0001, "env_id": "CartPole-v1", "state_space": 4, "action_space": 2}
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9bec14de42b9cf5aee1f746dbf8e794f2f9d87d2c1c2fc11db016a02eb18af0
3
+ size 2975
replay.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f98e7feed2b679be3ae6b7cd744d16d6e46de9c1c25f9e05f13013328814fe
3
+ size 10894
results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"env_id": "CartPole-v1", "mean_reward": 51.5, "n_evaluation_episodes": 20, "eval_datetime": "2022-07-11T17:10:29.239970"}