basil-ahmad commited on
Commit
677509b
1 Parent(s): 500da17

Push agent to the Hub

Browse files
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - CartPole-v1
4
+ - ppo
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - custom-implementation
8
+ - deep-rl-course
9
+ model-index:
10
+ - name: PPO
11
+ results:
12
+ - task:
13
+ type: reinforcement-learning
14
+ name: reinforcement-learning
15
+ dataset:
16
+ name: CartPole-v1
17
+ type: CartPole-v1
18
+ metrics:
19
+ - type: mean_reward
20
+ value: 236.40 +/- 117.78
21
+ name: mean_reward
22
+ verified: false
23
+ ---
24
+
25
+ # PPO Agent Playing CartPole-v1
26
+
27
+ This is a trained model of a PPO agent playing CartPole-v1.
28
+
29
+ # Hyperparameters
30
+ ```python
31
+ {'exp_name': '/root/.local/share/jupyter/runtime/kernel-b1251b3f-0560-43f9-a6d6-1db6758e48f1.json'
32
+ 'seed': 1
33
+ 'torch_deterministic': True
34
+ 'cuda': True
35
+ 'track': False
36
+ 'wandb_project_name': 'cleanRL'
37
+ 'wandb_entity': None
38
+ 'capture_video': False
39
+ 'env_id': 'CartPole-v1'
40
+ 'total_timesteps': 50000
41
+ 'learning_rate': 0.00025
42
+ 'num_envs': 4
43
+ 'num_steps': 128
44
+ 'anneal_lr': True
45
+ 'gae': True
46
+ 'gamma': 0.99
47
+ 'gae_lambda': 0.95
48
+ 'num_minibatches': 4
49
+ 'update_epochs': 4
50
+ 'norm_adv': True
51
+ 'clip_coef': 0.2
52
+ 'clip_vloss': True
53
+ 'ent_coef': 0.01
54
+ 'vf_coef': 0.5
55
+ 'max_grad_norm': 0.5
56
+ 'target_kl': None
57
+ 'repo_id': 'ThomasSimonini/ppo-CartPole-v1'
58
+ 'batch_size': 512
59
+ 'minibatch_size': 128}
60
+ ```
61
+
logs/events.out.tfevents.1712751487.28521fe26a6c.445.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8df2c4c917fc6e582ac518f884ed13d61ea83f0cfe618c615b21994c89cf13d9
3
+ size 108029
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:253f0e4b6304f0dc7378e7ccc2960aa8eaace3a0185975f83d3c3cb9877e6234
3
+ size 40466
replay.mp4 ADDED
Binary file (22.9 kB). View file
 
results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"env_id": "CartPole-v1", "mean_reward": 236.4, "std_reward": 117.78302084765869, "n_evaluation_episodes": 10, "eval_datetime": "2024-04-10T12:19:42.988392"}