Push agent to the Hub

README.md

@@ -0,0 +1,62 @@
+---
+tags:
+- CartPole-v1
+- ppo
+- deep-reinforcement-learning
+- reinforcement-learning
+- custom-implementation
+- deep-rl-course
+model-index:
+- name: PPO
+  results:
+  - task:
+      type: reinforcement-learning
+      name: reinforcement-learning
+    dataset:
+      name: CartPole-v1
+      type: CartPole-v1
+    metrics:
+    - type: mean_reward
+      value: 112.20 +/- 67.63
+      name: mean_reward
+      verified: false
+---
+
+  # PPO Agent Playing CartPole-v1
+
+  This is a trained model of a PPO agent playing CartPole-v1.
+
+  # Hyperparameters
+  ```python
+  {'exp_name': 'ppo-11'
+'gym_id': 'CartPole-v1'
+'env_id': 'CartPole-v1'
+'learning_rate': 0.00025
+'seed': 1
+'total_timesteps': 25000
+'torch_deterministic': True
+'cuda': True
+'track': False
+'wandb_project_name': 'ppo-implementation-details'
+'wandb_entity': None
+'capture_video': False
+'num_envs': 4
+'num_steps': 128
+'anneal_lr': True
+'gae': True
+'gamma': 0.99
+'gae_lambda': 0.95
+'num_minibatches': 4
+'update_epochs': 4
+'norm_adv': True
+'clip_coef': 0.2
+'clip_vloss': True
+'ent_coef': 0.01
+'vf_coef': 0.5
+'max_grad_norm': 0.5
+'target_kl': None
+'repo_id': 'juanzinser/ppo-CartPole-v1'
+'batch_size': 512
+'minibatch_size': 128}
+  ```
+  

logs/events.out.tfevents.1717294021.5ff29af29a63.34222.5

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

model.pt

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

results.json

@@ -0,0 +1 @@
+{"env_id": "CartPole-v1", "mean_reward": 112.2, "std_reward": 67.62514325308302, "n_evaluation_episodes": 10, "eval_datetime": "2024-06-02T02:07:55.266035"}