Push agent to the Hub
Browse files- README.md +58 -0
- model.pt +3 -0
- replay.mp4/ppo_hf/rl-video-episode-0.meta.json +1 -0
- replay.mp4/ppo_hf/rl-video-episode-0.mp4 +0 -0
- results.json +1 -0
README.md
ADDED
|
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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: 176.70 +/- 53.34
|
| 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': 'ppo_hf'
|
| 32 |
+
'env_id': 'CartPole-v1'
|
| 33 |
+
'learning_rate': 0.00025
|
| 34 |
+
'seed': 1
|
| 35 |
+
'total_timesteps': 25000
|
| 36 |
+
'torch_deterministic': True
|
| 37 |
+
'cuda': True
|
| 38 |
+
'capture_video': False
|
| 39 |
+
'num_envs': 4
|
| 40 |
+
'num_steps': 128
|
| 41 |
+
'anneal_lr': True
|
| 42 |
+
'gae': True
|
| 43 |
+
'gamma': 0.99
|
| 44 |
+
'gae_lambda': 0.95
|
| 45 |
+
'num_minibatches': 4
|
| 46 |
+
'update_epochs': 4
|
| 47 |
+
'norm_adv': True
|
| 48 |
+
'clip_coef': 0.2
|
| 49 |
+
'clip_loss': True
|
| 50 |
+
'ent_coef': 0.01
|
| 51 |
+
'vf_coef': 0.01
|
| 52 |
+
'max_grad_norm': 0.05
|
| 53 |
+
'target_kl': None
|
| 54 |
+
'repo_id': 'debisoft/ppo-CartPole-v1'
|
| 55 |
+
'batch_size': 512
|
| 56 |
+
'minibatch_size': 128}
|
| 57 |
+
```
|
| 58 |
+
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a7aee8ffbe3db82a9597343bb9266025f43cd64bc0e96bc4d6958274bb5a5d9
|
| 3 |
+
size 40731
|
replay.mp4/ppo_hf/rl-video-episode-0.meta.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"step_id": 0, "episode_id": 0, "content_type": "video/mp4", "encoder_version": {"backend": "ffmpeg", "version": "b'ffmpeg version 4.2.7-0ubuntu0.1 Copyright (c) 2000-2022 the FFmpeg developers\\nbuilt with gcc 9 (Ubuntu 9.4.0-1ubuntu1~20.04.1)\\nconfiguration: --prefix=/usr --extra-version=0ubuntu0.1 --toolchain=hardened --libdir=/usr/lib/x86_64-linux-gnu --incdir=/usr/include/x86_64-linux-gnu --arch=amd64 --enable-gpl --disable-stripping --enable-avresample --disable-filter=resample --enable-avisynth --enable-gnutls --enable-ladspa --enable-libaom --enable-libass --enable-libbluray --enable-libbs2b --enable-libcaca --enable-libcdio --enable-libcodec2 --enable-libflite --enable-libfontconfig --enable-libfreetype --enable-libfribidi --enable-libgme --enable-libgsm --enable-libjack --enable-libmp3lame --enable-libmysofa --enable-libopenjpeg --enable-libopenmpt --enable-libopus --enable-libpulse --enable-librsvg --enable-librubberband --enable-libshine --enable-libsnappy --enable-libsoxr --enable-libspeex --enable-libssh --enable-libtheora --enable-libtwolame --enable-libvidstab --enable-libvorbis --enable-libvpx --enable-libwavpack --enable-libwebp --enable-libx265 --enable-libxml2 --enable-libxvid --enable-libzmq --enable-libzvbi --enable-lv2 --enable-omx --enable-openal --enable-opencl --enable-opengl --enable-sdl2 --enable-libdc1394 --enable-libdrm --enable-libiec61883 --enable-nvenc --enable-chromaprint --enable-frei0r --enable-libx264 --enable-shared\\nlibavutil 56. 31.100 / 56. 31.100\\nlibavcodec 58. 54.100 / 58. 54.100\\nlibavformat 58. 29.100 / 58. 29.100\\nlibavdevice 58. 8.100 / 58. 8.100\\nlibavfilter 7. 57.100 / 7. 57.100\\nlibavresample 4. 0. 0 / 4. 0. 0\\nlibswscale 5. 5.100 / 5. 5.100\\nlibswresample 3. 5.100 / 3. 5.100\\nlibpostproc 55. 5.100 / 55. 5.100\\n'", "cmdline": ["ffmpeg", "-nostats", "-loglevel", "error", "-y", "-f", "rawvideo", "-s:v", "600x400", "-pix_fmt", "rgb24", "-framerate", "50", "-i", "-", "-vf", "scale=trunc(iw/2)*2:trunc(ih/2)*2", "-vcodec", "libx264", "-pix_fmt", "yuv420p", "-r", "50", "/tmp/tmp9esuf2_p/replay.mp4/ppo_hf/rl-video-episode-0.mp4"]}}
|
replay.mp4/ppo_hf/rl-video-episode-0.mp4
ADDED
|
Binary file (26.2 kB). View file
|
|
|
results.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"env_id": "CartPole-v1", "mean_reward": 176.7, "std_reward": 53.34425929751017, "n_evaluation_episodes": 10, "eval_datetime": "2025-10-25T08:03:49.770821"}
|