File size: 1,205 Bytes
f50803a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
---
tags:
- CartPole-v1
- ppo
- reinforcement-learning
- custom-implementation
- deep-rl-class
model-index:
- name: CleanRLTest
results:
- metrics:
- type: mean_reward
value: 239.10 +/- 117.52
name: mean_reward
task:
type: reinforcement-learning
name: reinforcement-learning
dataset:
name: CartPole-v1
type: CartPole-v1
---
# **PPO** Agent playing
This is a trained model of a ***PPO** agent playing CartPole-v1.
To learn to use this model and train yours check Unit 8 of the Deep Reinforcement Learning Class: https://github.com/huggingface/deep-rl-class/tree/main/unit5
{'exp_name': 'ppo', 'seed': 1, 'torch_deterministic': True, 'cuda': True, 'track': False, 'wandb_project_name': 'cleanRL', 'wandb_entity': None, 'capture_video': False, 'env_id': 'CartPole-v1', 'total_timesteps': 50000, 'learning_rate': 0.00025, '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, 'batch_size': 512, 'minibatch_size': 128}
Boom
|