Push agent to the Hub

README.md

@@ -17,7 +17,7 @@ model-index:
       type: CartPole-v1
     metrics:
     - type: mean_reward
-      value: 462.70 +/- 65.16
+      value: 225.90 +/- 71.33
       name: mean_reward
       verified: false
 ---
@@ -38,25 +38,25 @@ model-index:
 'wandb_entity': None
 'capture_video': False
 'env_id': 'CartPole-v1'
-'total_timesteps': 500000
-'learning_rate': 0.000225
-'num_envs': 4
-'num_steps': 256
+'total_timesteps': 900000
+'learning_rate': 0.000205
+'num_envs': 6
+'num_steps': 384
 'anneal_lr': True
 'gae': True
-'gamma': 0.99
-'gae_lambda': 0.95
+'gamma': 0.985
+'gae_lambda': 0.94
 'num_minibatches': 4
 'update_epochs': 4
 'norm_adv': True
-'clip_coef': 0.2
+'clip_coef': 0.225
 'clip_vloss': True
 'ent_coef': 0.01
 'vf_coef': 0.5
 'max_grad_norm': 0.5
 'target_kl': None
 'repo_id': 'kaljr/ppo-CartPole-v1'
-'batch_size': 1024
-'minibatch_size': 256}
+'batch_size': 2304
+'minibatch_size': 576}
   ```
   

logs/events.out.tfevents.1708907502.0d8f6edb1bcc.955.5

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

model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7c22652ee6081569258a8bf202953beb291814b195a2795cca2ab308bcffe697
+oid sha256:63d872a0867ed5d76310af353204cbe65f94feb5853ee9c365369088bb232521
 size 40466

results.json

@@ -1 +1 @@
-{"env_id": "CartPole-v1", "mean_reward": 462.7, "std_reward": 65.16448419192774, "n_evaluation_episodes": 10, "eval_datetime": "2024-02-26T00:29:16.406691"}
+{"env_id": "CartPole-v1", "mean_reward": 225.9, "std_reward": 71.32804497531109, "n_evaluation_episodes": 10, "eval_datetime": "2024-02-26T00:36:57.478614"}