benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest checkpoint epoch 1350
Browse files- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/config.json +48 -0
- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/history.json +0 -0
- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/metadata.json +57 -0
- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/policy.pt +3 -0
- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/progress.json +5 -0
- benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/value.pt +3 -0
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/config.json
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"env_id": "CartPole-v1",
|
| 3 |
+
"env_backend": "gymnasium",
|
| 4 |
+
"results_env_id": "",
|
| 5 |
+
"variant": "erc_trpo",
|
| 6 |
+
"total_timesteps": 3000000,
|
| 7 |
+
"max_episode_steps": 500,
|
| 8 |
+
"gamma": 0.98,
|
| 9 |
+
"gae_lambda": 0.95,
|
| 10 |
+
"max_kl": 1e-05,
|
| 11 |
+
"cg_iters": 10,
|
| 12 |
+
"cg_damping": 0.1,
|
| 13 |
+
"batch_size": 500,
|
| 14 |
+
"update_batch_size": 32,
|
| 15 |
+
"entropy_coef": 1e-08,
|
| 16 |
+
"policy_hidden": [
|
| 17 |
+
128
|
| 18 |
+
],
|
| 19 |
+
"value_hidden": [
|
| 20 |
+
128
|
| 21 |
+
],
|
| 22 |
+
"seed": 0,
|
| 23 |
+
"num_seeds": 3,
|
| 24 |
+
"log_interval": 10,
|
| 25 |
+
"eval_interval": 10,
|
| 26 |
+
"num_eval_episodes": 10,
|
| 27 |
+
"results_dir": "results",
|
| 28 |
+
"benchmark_id": "e3_3m",
|
| 29 |
+
"checkpoints_dir": "",
|
| 30 |
+
"use_buffer": false,
|
| 31 |
+
"buffer_clear_threshold": 195.0,
|
| 32 |
+
"buffer_capacity": 10000,
|
| 33 |
+
"device": "cpu",
|
| 34 |
+
"upload_to_hf": false,
|
| 35 |
+
"sync_hf_on_start": false,
|
| 36 |
+
"resume": true,
|
| 37 |
+
"save_every_epochs": 50,
|
| 38 |
+
"hf_upload_every_epochs": 0,
|
| 39 |
+
"hf_sync_interval_sec": 900,
|
| 40 |
+
"wipe_on_hyperparam_change": true,
|
| 41 |
+
"ppo_clip": 0.2,
|
| 42 |
+
"ppo_epochs": 10,
|
| 43 |
+
"ppo_lr": 0.0003,
|
| 44 |
+
"random_kl_std": 0.1,
|
| 45 |
+
"activation": "relu",
|
| 46 |
+
"value_lr": 0.01,
|
| 47 |
+
"value_epochs": 20
|
| 48 |
+
}
|
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/history.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/metadata.json
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"env_id": "CartPole-v1",
|
| 3 |
+
"discrete_actions": true,
|
| 4 |
+
"device": "cpu",
|
| 5 |
+
"policy_hidden": [
|
| 6 |
+
128
|
| 7 |
+
],
|
| 8 |
+
"value_hidden": [
|
| 9 |
+
128
|
| 10 |
+
],
|
| 11 |
+
"training": {
|
| 12 |
+
"total_timesteps": 3000000,
|
| 13 |
+
"gamma": 0.98,
|
| 14 |
+
"entropy_coef": 1e-08,
|
| 15 |
+
"seed": 0,
|
| 16 |
+
"batch_size": 500
|
| 17 |
+
},
|
| 18 |
+
"progress": {
|
| 19 |
+
"epoch": 1350,
|
| 20 |
+
"timesteps": 675000
|
| 21 |
+
},
|
| 22 |
+
"metrics": {
|
| 23 |
+
"epoch": 1349,
|
| 24 |
+
"timesteps": 675000,
|
| 25 |
+
"objective": 5.607032653642818e-05,
|
| 26 |
+
"surrogate_improve": 5.607032653642818e-05,
|
| 27 |
+
"surrogate": 5.607032653642818e-05,
|
| 28 |
+
"kl": 5.49876676814165e-06,
|
| 29 |
+
"entropy_mean": 0.6123444437980652,
|
| 30 |
+
"kl_budget_bonus": 6.123444437980652e-09,
|
| 31 |
+
"effective_kl": 5.49876676814165e-06,
|
| 32 |
+
"line_search_backtracks": 4,
|
| 33 |
+
"update_success": 1.0,
|
| 34 |
+
"kl_relaxation": 6.123444437980652e-09,
|
| 35 |
+
"eval_return_mean": 115.6,
|
| 36 |
+
"eval_return_std": 42.38207168131355
|
| 37 |
+
},
|
| 38 |
+
"variant": "erc_trpo",
|
| 39 |
+
"uses_entropy": false,
|
| 40 |
+
"uses_buffer": false,
|
| 41 |
+
"paper": {
|
| 42 |
+
"title": "Trust region policy optimization via entropy regularization for Kullback\u2013Leibler divergence constraint",
|
| 43 |
+
"authors": "Haotian Xu, Junyu Xuan, Guangquan Zhang, Jie Lu",
|
| 44 |
+
"year": 2024,
|
| 45 |
+
"venue": "Neurocomputing 589:127716",
|
| 46 |
+
"url": "https://doi.org/10.1016/j.neucom.2024.127716"
|
| 47 |
+
},
|
| 48 |
+
"secondary_papers": [
|
| 49 |
+
{
|
| 50 |
+
"title": "Trust Region Policy Optimization",
|
| 51 |
+
"authors": "John Schulman, Sergey Levine, Philipp Moritz, Michael I. Jordan, Pieter Abbeel",
|
| 52 |
+
"year": 2015,
|
| 53 |
+
"venue": "ICML",
|
| 54 |
+
"url": ""
|
| 55 |
+
}
|
| 56 |
+
]
|
| 57 |
+
}
|
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d97f37e45df229ab0d986788e534a8f75f3c15755372ba1a5e17461cec4ed058
|
| 3 |
+
size 5971
|
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/progress.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 1350,
|
| 3 |
+
"timesteps": 675000,
|
| 4 |
+
"history_len": 1350
|
| 5 |
+
}
|
benchmark_e3_3m/CartPole-v1/erc_trpo/seed_0/latest/value.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d084c76e496e7c9e05fe2e68c9d0e2851724ae7c1b491eafac12959c98ab37c9
|
| 3 |
+
size 5449
|