danielnbarbosa commited on
Commit
a7c69d9
·
verified ·
1 Parent(s): 2ad14b7

Initial commit

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: stable-baselines3
3
+ tags:
4
+ - CartPole-v1
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - stable-baselines3
8
+ model-index:
9
+ - name: PPO
10
+ results:
11
+ - task:
12
+ type: reinforcement-learning
13
+ name: reinforcement-learning
14
+ dataset:
15
+ name: CartPole-v1
16
+ type: CartPole-v1
17
+ metrics:
18
+ - type: mean_reward
19
+ value: 500.00 +/- 0.00
20
+ name: mean_reward
21
+ verified: false
22
+ ---
23
+
24
+ # **PPO** Agent playing **CartPole-v1**
25
+ This is a trained model of a **PPO** agent playing **CartPole-v1**
26
+ using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3)
27
+ and the [RL Zoo](https://github.com/DLR-RM/rl-baselines3-zoo).
28
+
29
+ The RL Zoo is a training framework for Stable Baselines3
30
+ reinforcement learning agents,
31
+ with hyperparameter optimization and pre-trained agents included.
32
+
33
+ ## Usage (with SB3 RL Zoo)
34
+
35
+ RL Zoo: https://github.com/DLR-RM/rl-baselines3-zoo<br/>
36
+ SB3: https://github.com/DLR-RM/stable-baselines3<br/>
37
+ SB3 Contrib: https://github.com/Stable-Baselines-Team/stable-baselines3-contrib
38
+ SBX (SB3 + Jax): https://github.com/araffin/sbx
39
+
40
+ Install the RL Zoo (with SB3 and SB3-Contrib):
41
+ ```bash
42
+ pip install rl_zoo3
43
+ ```
44
+
45
+ ```
46
+ # Download model and save it into the logs/ folder
47
+ python -m rl_zoo3.load_from_hub --algo ppo --env CartPole-v1 -orga danielnbarbosa -f logs/
48
+ python -m rl_zoo3.enjoy --algo ppo --env CartPole-v1 -f logs/
49
+ ```
50
+
51
+ If you installed the RL Zoo3 via pip (`pip install rl_zoo3`), from anywhere you can do:
52
+ ```
53
+ python -m rl_zoo3.load_from_hub --algo ppo --env CartPole-v1 -orga danielnbarbosa -f logs/
54
+ python -m rl_zoo3.enjoy --algo ppo --env CartPole-v1 -f logs/
55
+ ```
56
+
57
+ ## Training (with the RL Zoo)
58
+ ```
59
+ python -m rl_zoo3.train --algo ppo --env CartPole-v1 -f logs/
60
+ # Upload the model and generate video (when possible)
61
+ python -m rl_zoo3.push_to_hub --algo ppo --env CartPole-v1 -f logs/ -orga danielnbarbosa
62
+ ```
63
+
64
+ ## Hyperparameters
65
+ ```python
66
+ OrderedDict([('batch_size', 256),
67
+ ('clip_range', 'lin_0.2'),
68
+ ('ent_coef', 0.0),
69
+ ('gae_lambda', 0.8),
70
+ ('gamma', 0.98),
71
+ ('learning_rate', 'lin_0.001'),
72
+ ('n_envs', 8),
73
+ ('n_epochs', 20),
74
+ ('n_steps', 32),
75
+ ('n_timesteps', 100000.0),
76
+ ('policy', 'MlpPolicy'),
77
+ ('normalize', False)])
78
+ ```
79
+
80
+ # Environment Arguments
81
+ ```python
82
+ {'render_mode': 'rgb_array'}
83
+ ```
args.yml ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object/apply:collections.OrderedDict
2
+ - - - algo
3
+ - ppo
4
+ - - conf_file
5
+ - null
6
+ - - device
7
+ - auto
8
+ - - env
9
+ - CartPole-v1
10
+ - - env_kwargs
11
+ - null
12
+ - - eval_env_kwargs
13
+ - null
14
+ - - eval_episodes
15
+ - 5
16
+ - - eval_freq
17
+ - 25000
18
+ - - gym_packages
19
+ - []
20
+ - - hyperparams
21
+ - null
22
+ - - log_folder
23
+ - logs
24
+ - - log_interval
25
+ - -1
26
+ - - max_total_trials
27
+ - null
28
+ - - n_eval_envs
29
+ - 1
30
+ - - n_evaluations
31
+ - null
32
+ - - n_jobs
33
+ - 1
34
+ - - n_startup_trials
35
+ - 10
36
+ - - n_timesteps
37
+ - -1
38
+ - - n_trials
39
+ - 500
40
+ - - no_optim_plots
41
+ - false
42
+ - - num_threads
43
+ - -1
44
+ - - optimization_log_path
45
+ - null
46
+ - - optimize_hyperparameters
47
+ - false
48
+ - - progress
49
+ - false
50
+ - - pruner
51
+ - median
52
+ - - sampler
53
+ - tpe
54
+ - - save_freq
55
+ - -1
56
+ - - save_replay_buffer
57
+ - false
58
+ - - seed
59
+ - 11746133
60
+ - - storage
61
+ - null
62
+ - - study_name
63
+ - null
64
+ - - tensorboard_log
65
+ - runs/CartPole-v1__ppo__11746133__1742912977
66
+ - - track
67
+ - true
68
+ - - trained_agent
69
+ - ''
70
+ - - trial_id
71
+ - null
72
+ - - truncate_last_trajectory
73
+ - true
74
+ - - uuid
75
+ - false
76
+ - - vec_env
77
+ - dummy
78
+ - - verbose
79
+ - 1
80
+ - - wandb_entity
81
+ - null
82
+ - - wandb_project_name
83
+ - sb3
84
+ - - wandb_tags
85
+ - []
config.yml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object/apply:collections.OrderedDict
2
+ - - - batch_size
3
+ - 256
4
+ - - clip_range
5
+ - lin_0.2
6
+ - - ent_coef
7
+ - 0.0
8
+ - - gae_lambda
9
+ - 0.8
10
+ - - gamma
11
+ - 0.98
12
+ - - learning_rate
13
+ - lin_0.001
14
+ - - n_envs
15
+ - 8
16
+ - - n_epochs
17
+ - 20
18
+ - - n_steps
19
+ - 32
20
+ - - n_timesteps
21
+ - 100000.0
22
+ - - policy
23
+ - MlpPolicy
env_kwargs.yml ADDED
@@ -0,0 +1 @@
 
 
1
+ render_mode: rgb_array
ppo-CartPole-v1.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f82095022c437fd3b89d931a3b52a7fb6336b3656b6049b15dfe3ce290b2c4f8
3
+ size 146248
ppo-CartPole-v1/_stable_baselines3_version ADDED
@@ -0,0 +1 @@
 
 
1
+ 2.6.0a2
ppo-CartPole-v1/data ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "policy_class": {
3
+ ":type:": "<class 'abc.ABCMeta'>",
4
+ ":serialized:": "gAWVOwAAAAAAAACMIXN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi5wb2xpY2llc5SMEUFjdG9yQ3JpdGljUG9saWN5lJOULg==",
5
+ "__module__": "stable_baselines3.common.policies",
6
+ "__firstlineno__": 416,
7
+ "__doc__": "\nPolicy class for actor-critic algorithms (has both policy and value prediction).\nUsed by A2C, PPO and the likes.\n\n:param observation_space: Observation space\n:param action_space: Action space\n:param lr_schedule: Learning rate schedule (could be constant)\n:param net_arch: The specification of the policy and value networks.\n:param activation_fn: Activation function\n:param ortho_init: Whether to use or not orthogonal initialization\n:param use_sde: Whether to use State Dependent Exploration or not\n:param log_std_init: Initial value for the log standard deviation\n:param full_std: Whether to use (n_features x n_actions) parameters\n for the std instead of only (n_features,) when using gSDE\n:param use_expln: Use ``expln()`` function instead of ``exp()`` to ensure\n a positive standard deviation (cf paper). It allows to keep variance\n above zero and prevent it from growing too fast. In practice, ``exp()`` is usually enough.\n:param squash_output: Whether to squash the output using a tanh function,\n this allows to ensure boundaries when using gSDE.\n:param features_extractor_class: Features extractor to use.\n:param features_extractor_kwargs: Keyword arguments\n to pass to the features extractor.\n:param share_features_extractor: If True, the features extractor is shared between the policy and value networks.\n:param normalize_images: Whether to normalize images or not,\n dividing by 255.0 (True by default)\n:param optimizer_class: The optimizer to use,\n ``th.optim.Adam`` by default\n:param optimizer_kwargs: Additional keyword arguments,\n excluding the learning rate, to pass to the optimizer\n",
8
+ "__init__": "<function ActorCriticPolicy.__init__ at 0x11ccf87c0>",
9
+ "_get_constructor_parameters": "<function ActorCriticPolicy._get_constructor_parameters at 0x11ccf8860>",
10
+ "reset_noise": "<function ActorCriticPolicy.reset_noise at 0x11ccf8900>",
11
+ "_build_mlp_extractor": "<function ActorCriticPolicy._build_mlp_extractor at 0x11ccf89a0>",
12
+ "_build": "<function ActorCriticPolicy._build at 0x11ccf8a40>",
13
+ "forward": "<function ActorCriticPolicy.forward at 0x11ccf8ae0>",
14
+ "extract_features": "<function ActorCriticPolicy.extract_features at 0x11ccf8b80>",
15
+ "_get_action_dist_from_latent": "<function ActorCriticPolicy._get_action_dist_from_latent at 0x11ccf8c20>",
16
+ "_predict": "<function ActorCriticPolicy._predict at 0x11ccf8cc0>",
17
+ "evaluate_actions": "<function ActorCriticPolicy.evaluate_actions at 0x11ccf8d60>",
18
+ "get_distribution": "<function ActorCriticPolicy.get_distribution at 0x11ccf8e00>",
19
+ "predict_values": "<function ActorCriticPolicy.predict_values at 0x11ccf8ea0>",
20
+ "__static_attributes__": [
21
+ "action_dist",
22
+ "action_net",
23
+ "activation_fn",
24
+ "dist_kwargs",
25
+ "features_dim",
26
+ "features_extractor",
27
+ "log_std",
28
+ "log_std_init",
29
+ "mlp_extractor",
30
+ "net_arch",
31
+ "optimizer",
32
+ "ortho_init",
33
+ "pi_features_extractor",
34
+ "share_features_extractor",
35
+ "use_sde",
36
+ "value_net",
37
+ "vf_features_extractor"
38
+ ],
39
+ "__abstractmethods__": "frozenset()",
40
+ "_abc_impl": "<_abc._abc_data object at 0x11ccf1c40>"
41
+ },
42
+ "verbose": 1,
43
+ "policy_kwargs": {},
44
+ "num_timesteps": 100096,
45
+ "_total_timesteps": 100000,
46
+ "_num_timesteps_at_start": 0,
47
+ "seed": 0,
48
+ "action_noise": null,
49
+ "start_time": 1742912979988956000,
50
+ "learning_rate": {
51
+ ":type:": "<class 'function'>",
52
+ ":serialized:": "gAWVdwMAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLAksTQw4+AZUAVQBUAS0FAAAkAJSMZApQcm9ncmVzcyB3aWxsIGRlY3JlYXNlIGZyb20gMSAoYmVnaW5uaW5nKSB0byAwCjpwYXJhbSBwcm9ncmVzc19yZW1haW5pbmc6IChmbG9hdCkKOnJldHVybjogKGZsb2F0KQqUhZQpjBJwcm9ncmVzc19yZW1haW5pbmeUhZSMQi9Vc2Vycy9kYW5pZWwvc3JjL2dpdGh1Yi9ETFItUk0vcmwtYmFzZWxpbmVzMy16b28vcmxfem9vMy91dGlscy5weZSMBGZ1bmOUjB1saW5lYXJfc2NoZWR1bGUuPGxvY2Fscz4uZnVuY5RNMQFDEPiAAPAMABAioE7RDzLQCDKUQwCUjA5pbml0aWFsX3ZhbHVlX5SFlCl0lFKUfZQojAtfX3BhY2thZ2VfX5SMB3JsX3pvbzOUjAhfX25hbWVfX5SMDXJsX3pvbzMudXRpbHOUjAhfX2ZpbGVfX5SMQi9Vc2Vycy9kYW5pZWwvc3JjL2dpdGh1Yi9ETFItUk0vcmwtYmFzZWxpbmVzMy16b28vcmxfem9vMy91dGlscy5weZR1Tk5oAIwQX21ha2VfZW1wdHlfY2VsbJSTlClSlIWUdJRSlGgAjBJfZnVuY3Rpb25fc2V0c3RhdGWUk5RoIn2UfZQoaBmMBGZ1bmOUjAxfX3F1YWxuYW1lX1+UjB1saW5lYXJfc2NoZWR1bGUuPGxvY2Fscz4uZnVuY5SMD19fYW5ub3RhdGlvbnNfX5R9lCiMEnByb2dyZXNzX3JlbWFpbmluZ5SMCGJ1aWx0aW5zlIwFZmxvYXSUk5SMBnJldHVybpRoL3WMDl9fa3dkZWZhdWx0c19flE6MDF9fZGVmYXVsdHNfX5ROjApfX21vZHVsZV9flGgajAdfX2RvY19flGgJjAtfX2Nsb3N1cmVfX5RoAIwKX21ha2VfY2VsbJSTlEc/UGJN0vGp/IWUUpSFlIwXX2Nsb3VkcGlja2xlX3N1Ym1vZHVsZXOUXZSMC19fZ2xvYmFsc19flH2UdYaUhlIwLg=="
53
+ },
54
+ "tensorboard_log": "runs/CartPole-v1__ppo__11746133__1742912977/CartPole-v1",
55
+ "_last_obs": null,
56
+ "_last_episode_starts": {
57
+ ":type:": "<class 'numpy.ndarray'>",
58
+ ":serialized:": "gAWVfAAAAAAAAACME251bXB5Ll9jb3JlLm51bWVyaWOUjAtfZnJvbWJ1ZmZlcpSTlCiWCAAAAAAAAAAAAAAAAAAAAJSMBW51bXB5lIwFZHR5cGWUk5SMAmIxlImIh5RSlChLA4wBfJROTk5K/////0r/////SwB0lGJLCIWUjAFDlHSUUpQu"
59
+ },
60
+ "_last_original_obs": null,
61
+ "_episode_num": 0,
62
+ "use_sde": false,
63
+ "sde_sample_freq": -1,
64
+ "_current_progress_remaining": -0.0009600000000000719,
65
+ "_stats_window_size": 100,
66
+ "ep_info_buffer": {
67
+ ":type:": "<class 'collections.deque'>",
68
+ ":serialized:": "gAWVRAwAAAAAAACMC2NvbGxlY3Rpb25zlIwFZGVxdWWUk5QpS2SGlFKUKH2UKIwBcpRHQH9AAAAAAACMAWyUTfQBjAF0lEdAGwCDmKZUk3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQBttvCMxXXB1fZQoaAZHQH9AAAAAAABoB030AWgIR0AbwjzI3irDdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAG9hvze40/HV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQBvyZBsyi251fZQoaAZHQH9AAAAAAABoB030AWgIR0Ab+lhw2l2vdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAHAyXUpd8iXV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQBxLZnL7oB91fZQoaAZHQH9AAAAAAABoB030AWgIR0AcuMvRJEpidX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAHSmCAc1fmnV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQB18WCVbA1x1fZQoaAZHQH9AAAAAAABoB030AWgIR0Adfd+G47RwdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAHZe+23KB/nV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQB21JUYKpkx1fZQoaAZHQH9AAAAAAABoB030AWgIR0AdsxZdOZb7dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAHgYO2AoXsXV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQB6BAWznied1fZQoaAZHQH9AAAAAAABoB030AWgIR0Ae8e8wpON6dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAH0UL2HtWuHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQB9GwFC9h7V1fZQoaAZHQH9AAAAAAABoB030AWgIR0AfYU9IPK+0dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAH33WWhRIjHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQB970z0pVjt1fZQoaAZHQH9AAAAAAABoB030AWgIR0AfzSNOuaF3dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIBJ+DvmYB3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCBJCrtE5Qx1fZQoaAZHQH9AAAAAAABoB030AWgIR0AgdBX0XgtOdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIH6cAiml7HV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCCLr/sE7nx1fZQoaAZHQH9AAAAAAABoB030AWgIR0Agj5C4SYgJdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAII6DXe3x4XV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCC49V3ljmV1fZQoaAZHQH9AAAAAAABoB030AWgIR0Ag7p5/smfHdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAISYqG1x82XV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCFP8hs67ul1fZQoaAZHQH9AAAAAAABoB030AWgIR0AhUL1mJ3xGdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIV3hn8Koh3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCFscbR4QjF1fZQoaAZHQH9AAAAAAABoB030AWgIR0Aha2uPmxMWdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIZQtJ4B3inV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCHABYFJQLx1fZQoaAZHQH9AAAAAAABoB030AWgIR0Ah9s6aLGaQdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIh/XwsoUjHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCIq42CNCJJ1fZQoaAZHQH9AAAAAAABoB030AWgIR0AiONGViWmhdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIjyiudPLxXV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCJFx6v7m+11fZQoaAZHQH9AAAAAAABoB030AWgIR0Aib8Ti83+/dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAIpqp97Wuo3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCNUdmxt52R1fZQoaAZHQH9AAAAAAABoB030AWgIR0AjfkXDWK/EdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAI4kwWWQfZHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCOM2DQJHAh1fZQoaAZHQH9AAAAAAABoB030AWgIR0Ajm6eXiR4hdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAI5qdYnv2G3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCPFtGd7OVx1fZQoaAZHQH9AAAAAAABoB030AWgIR0Aj/CtRvWH2dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJDNNzr/sFHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCRdd/rjYI11fZQoaAZHQH9AAAAAAABoB030AWgIR0AkXmeUY8+zdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJGwLux8lX3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCR5++dsi0R1fZQoaAZHQH9AAAAAAABoB030AWgIR0AkePDpC8e0dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJKJL26ClJ3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCTNl9Sde6Z1fZQoaAZHQH9AAAAAAABoB030AWgIR0AlBdHlOoHcdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJTAWac7Qs3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCU7gn+hoM91fZQoaAZHQH9AAAAAAABoB030AWgIR0AlSKgqVhTgdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJUyBTXJ5mnV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCVVu+AVfu11fZQoaAZHQH9AAAAAAABoB030AWgIR0AlfL5AQg9vdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJcFl9Sde6nV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCYFqDbrTph1fZQoaAZHQH9AAAAAAABoB030AWgIR0AmOMd92HLzdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJjnH3lCCz3V9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCZKUiY9gWt1fZQoaAZHQH9AAAAAAABoB030AWgIR0AmWxlg+hXbdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJlqGDcuannV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCaOYtxuKoB1fZQoaAZHQH9AAAAAAABoB030AWgIR0Am0hXbM5fddX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJxZZbILgGnV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCdLr/sE7nx1fZQoaAZHQH9AAAAAAABoB030AWgIR0AnTQoCuEEldX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJ12OAAhjfHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCduP7vXsgN1fZQoaAZHQH9AAAAAAABoB030AWgIR0AnbWK/EfkndX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAJ6BfrrxAjnV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCfVJOFg2Ih1fZQoaAZHQH9AAAAAAABoB030AWgIR0AoF7gKnei0dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAKErB0p3HJnV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQChZDArQPZt1fZQoaAZHQH9AAAAAAABoB030AWgIR0AoaZwXIlt1dX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAKG2x6fJ3gXV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQChssFt8/lh1fZQoaAZHQH9AAAAAAABoB030AWgIR0Aon9gnc+JQdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAKOHGbTc7AHV9lChoBkdAf0AAAAAAAGgHTfQBaAhHQCkk/0NBnjB1fZQoaAZHQH9AAAAAAABoB030AWgIR0ApV+KCQLeAdX2UKGgGR0B/QAAAAAAAaAdN9AFoCEdAKeF9Brvb5HVlLg=="
69
+ },
70
+ "ep_success_buffer": {
71
+ ":type:": "<class 'collections.deque'>",
72
+ ":serialized:": "gAWVIAAAAAAAAACMC2NvbGxlY3Rpb25zlIwFZGVxdWWUk5QpS2SGlFKULg=="
73
+ },
74
+ "_n_updates": 7820,
75
+ "observation_space": {
76
+ ":type:": "<class 'gymnasium.spaces.box.Box'>",
77
+ ":serialized:": "gAWV/wEAAAAAAACMFGd5bW5hc2l1bS5zcGFjZXMuYm94lIwDQm94lJOUKYGUfZQojAVkdHlwZZSMBW51bXB5lIwFZHR5cGWUk5SMAmY0lImIh5RSlChLA4wBPJROTk5K/////0r/////SwB0lGKMBl9zaGFwZZRLBIWUjANsb3eUjBNudW1weS5fY29yZS5udW1lcmljlIwLX2Zyb21idWZmZXKUk5QolhAAAAAAAAAAmpmZwAAAgP9Qd9a+AACA/5RoC0sEhZSMAUOUdJRSlIwNYm91bmRlZF9iZWxvd5RoEyiWBAAAAAAAAAABAAEAlGgIjAJiMZSJiIeUUpQoSwOMAXyUTk5OSv////9K/////0sAdJRiSwSFlGgWdJRSlIwEaGlnaJRoEyiWEAAAAAAAAACamZlAAACAf1B31j4AAIB/lGgLSwSFlGgWdJRSlIwNYm91bmRlZF9hYm92ZZRoEyiWBAAAAAAAAAABAAEAlGgdSwSFlGgWdJRSlIwIbG93X3JlcHKUjDFbLTQuOCAgICAgICAgICAgICAgIC1pbmYgLTAuNDE4ODc5MDMgICAgICAgIC1pbmZdlIwJaGlnaF9yZXBylIwtWzQuOCAgICAgICAgICAgICAgIGluZiAwLjQxODg3OTAzICAgICAgICBpbmZdlIwKX25wX3JhbmRvbZROdWIu",
78
+ "dtype": "float32",
79
+ "_shape": [
80
+ 4
81
+ ],
82
+ "low": "[-4.8 -inf -0.41887903 -inf]",
83
+ "bounded_below": "[ True False True False]",
84
+ "high": "[4.8 inf 0.41887903 inf]",
85
+ "bounded_above": "[ True False True False]",
86
+ "low_repr": "[-4.8 -inf -0.41887903 -inf]",
87
+ "high_repr": "[4.8 inf 0.41887903 inf]",
88
+ "_np_random": null
89
+ },
90
+ "action_space": {
91
+ ":type:": "<class 'gymnasium.spaces.discrete.Discrete'>",
92
+ ":serialized:": "gAWVtQIAAAAAAACMGWd5bW5hc2l1bS5zcGFjZXMuZGlzY3JldGWUjAhEaXNjcmV0ZZSTlCmBlH2UKIwBbpSMFm51bXB5Ll9jb3JlLm11bHRpYXJyYXmUjAZzY2FsYXKUk5SMBW51bXB5lIwFZHR5cGWUk5SMAmk4lImIh5RSlChLA4wBPJROTk5K/////0r/////SwB0lGJDCAIAAAAAAAAAlIaUUpSMBXN0YXJ0lGgIaA5DCAAAAAAAAAAAlIaUUpSMBl9zaGFwZZQpjAVkdHlwZZRoC4wCaTiUiYiHlFKUKEsDaA9OTk5K/////0r/////SwB0lGKMCl9ucF9yYW5kb22UjBRudW1weS5yYW5kb20uX3BpY2tsZZSMEF9fZ2VuZXJhdG9yX2N0b3KUk5RoH4wUX19iaXRfZ2VuZXJhdG9yX2N0b3KUk5SME251bXB5LnJhbmRvbS5fcGNnNjSUjAVQQ0c2NJSTlIWUUpR9lCiMDWJpdF9nZW5lcmF0b3KUjAVQQ0c2NJSMBXN0YXRllH2UKGgsihDjYZWmt15YCS1Fllk0taEajANpbmOUihCpc3hEvDOBWIIa9zrb2o1BdYwKaGFzX3VpbnQzMpRLAIwIdWludGVnZXKUSwB1jBpudW1weS5yYW5kb20uYml0X2dlbmVyYXRvcpSMG19fcHl4X3VucGlja2xlX1NlZWRTZXF1ZW5jZZSTlGgxjAxTZWVkU2VxdWVuY2WUk5RKIqLqA06HlFKUKEsASwCME251bXB5Ll9jb3JlLm51bWVyaWOUjAtfZnJvbWJ1ZmZlcpSTlCiWEAAAAAAAAAAH60D+Njo2T50Ask6neprIlGgLjAJ1NJSJiIeUUpQoSwNoD05OTkr/////Sv////9LAHSUYksEhZSMAUOUdJRSlEsEKXSUYoaUYoWUUpR1Yi4=",
93
+ "n": "2",
94
+ "start": "0",
95
+ "_shape": [],
96
+ "dtype": "int64",
97
+ "_np_random": "Generator(PCG64)"
98
+ },
99
+ "n_envs": 1,
100
+ "n_steps": 32,
101
+ "gamma": 0.98,
102
+ "gae_lambda": 0.8,
103
+ "ent_coef": 0.0,
104
+ "vf_coef": 0.5,
105
+ "max_grad_norm": 0.5,
106
+ "rollout_buffer_class": {
107
+ ":type:": "<class 'abc.ABCMeta'>",
108
+ ":serialized:": "gAWVNgAAAAAAAACMIHN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi5idWZmZXJzlIwNUm9sbG91dEJ1ZmZlcpSTlC4=",
109
+ "__module__": "stable_baselines3.common.buffers",
110
+ "__firstlineno__": 343,
111
+ "__annotations__": "{'observations': <class 'numpy.ndarray'>, 'actions': <class 'numpy.ndarray'>, 'rewards': <class 'numpy.ndarray'>, 'advantages': <class 'numpy.ndarray'>, 'returns': <class 'numpy.ndarray'>, 'episode_starts': <class 'numpy.ndarray'>, 'log_probs': <class 'numpy.ndarray'>, 'values': <class 'numpy.ndarray'>}",
112
+ "__doc__": "\nRollout buffer used in on-policy algorithms like A2C/PPO.\nIt corresponds to ``buffer_size`` transitions collected\nusing the current policy.\nThis experience will be discarded after the policy update.\nIn order to use PPO objective, we also store the current value of each state\nand the log probability of each taken action.\n\nThe term rollout here refers to the model-free notion and should not\nbe used with the concept of rollout used in model-based RL or planning.\nHence, it is only involved in policy and value function training but not action selection.\n\n:param buffer_size: Max number of element in the buffer\n:param observation_space: Observation space\n:param action_space: Action space\n:param device: PyTorch device\n:param gae_lambda: Factor for trade-off of bias vs variance for Generalized Advantage Estimator\n Equivalent to classic advantage when set to 1.\n:param gamma: Discount factor\n:param n_envs: Number of parallel environments\n",
113
+ "__init__": "<function RolloutBuffer.__init__ at 0x119d5f380>",
114
+ "reset": "<function RolloutBuffer.reset at 0x119d5f420>",
115
+ "compute_returns_and_advantage": "<function RolloutBuffer.compute_returns_and_advantage at 0x119d5f4c0>",
116
+ "add": "<function RolloutBuffer.add at 0x119d5f600>",
117
+ "get": "<function RolloutBuffer.get at 0x119d5f6a0>",
118
+ "_get_samples": "<function RolloutBuffer._get_samples at 0x119d5f740>",
119
+ "__static_attributes__": [
120
+ "actions",
121
+ "advantages",
122
+ "episode_starts",
123
+ "full",
124
+ "gae_lambda",
125
+ "gamma",
126
+ "generator_ready",
127
+ "log_probs",
128
+ "observations",
129
+ "returns",
130
+ "rewards",
131
+ "values"
132
+ ],
133
+ "__abstractmethods__": "frozenset()",
134
+ "_abc_impl": "<_abc._abc_data object at 0x119d59640>"
135
+ },
136
+ "rollout_buffer_kwargs": {},
137
+ "batch_size": 256,
138
+ "n_epochs": 20,
139
+ "clip_range": {
140
+ ":type:": "<class 'function'>",
141
+ ":serialized:": "gAWVjwcAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLBUsTQyY+AZUAWwEAAAAAAAAAAFQBIgBVADUBAAAAAAAANQEAAAAAAAAkAJROhZSMBWZsb2F0lIWUjBJwcm9ncmVzc19yZW1haW5pbmeUhZSMgC9Vc2Vycy9kYW5pZWwvLmxvY2FsL3NoYXJlL3ZpcnR1YWxlbnZzL3JsLWJhc2VsaW5lczMtem9vLV83Wi01OF9vL2xpYi9weXRob24zLjEzL3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lIwIPGxhbWJkYT6UjCFnZXRfc2NoZWR1bGVfZm4uPGxvY2Fscz4uPGxhbWJkYT6US2JDEPiAAKRlqU7QO03TLE7UJk+UQwCUjA52YWx1ZV9zY2hlZHVsZZSFlCl0lFKUfZQojAtfX3BhY2thZ2VfX5SMGHN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbpSMCF9fbmFtZV9flIwec3RhYmxlX2Jhc2VsaW5lczMuY29tbW9uLnV0aWxzlIwIX19maWxlX1+UjIAvVXNlcnMvZGFuaWVsLy5sb2NhbC9zaGFyZS92aXJ0dWFsZW52cy9ybC1iYXNlbGluZXMzLXpvby1fN1otNThfby9saWIvcHl0aG9uMy4xMy9zaXRlLXBhY2thZ2VzL3N0YWJsZV9iYXNlbGluZXMzL2NvbW1vbi91dGlscy5weZR1Tk5oAIwQX21ha2VfZW1wdHlfY2VsbJSTlClSlIWUdJRSlGgAjBJfZnVuY3Rpb25fc2V0c3RhdGWUk5RoI32UfZQoaBqMCDxsYW1iZGE+lIwMX19xdWFsbmFtZV9flIwhZ2V0X3NjaGVkdWxlX2ZuLjxsb2NhbHM+LjxsYW1iZGE+lIwPX19hbm5vdGF0aW9uc19flH2UjA5fX2t3ZGVmYXVsdHNfX5ROjAxfX2RlZmF1bHRzX1+UTowKX19tb2R1bGVfX5RoG4wHX19kb2NfX5ROjAtfX2Nsb3N1cmVfX5RoAIwKX21ha2VfY2VsbJSTlGgCKGgHKEsBSwBLAEsBSwVLE0MmPgGVAFsBAAAAAAAAAABUASIAVQA1AQAAAAAAADUBAAAAAAAAJACUToWUaAqFlGgMhZRoDowIPGxhbWJkYT6UaBBLYkMQ+IAApGWpTtA7TdMsTtQmT5RoEmgThZQpdJRSlH2UKGgYjBhzdGFibGVfYmFzZWxpbmVzMy5jb21tb26UaBqMHnN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi51dGlsc5RoHIyAL1VzZXJzL2RhbmllbC8ubG9jYWwvc2hhcmUvdmlydHVhbGVudnMvcmwtYmFzZWxpbmVzMy16b28tXzdaLTU4X28vbGliL3B5dGhvbjMuMTMvc2l0ZS1wYWNrYWdlcy9zdGFibGVfYmFzZWxpbmVzMy9jb21tb24vdXRpbHMucHmUdU5OaB8pUpSFlHSUUpRoJWhEfZR9lChoGowIPGxhbWJkYT6UaCmMIWdldF9zY2hlZHVsZV9mbi48bG9jYWxzPi48bGFtYmRhPpRoK32UaC1OaC5OaC9oP2gwTmgxaDNoAihoByhLAUsASwBLAUsCSxNDDj4BlQBVAFQBLQUAACQAlIxkClByb2dyZXNzIHdpbGwgZGVjcmVhc2UgZnJvbSAxIChiZWdpbm5pbmcpIHRvIDAKOnBhcmFtIHByb2dyZXNzX3JlbWFpbmluZzogKGZsb2F0KQo6cmV0dXJuOiAoZmxvYXQpCpSFlCloDIWUjEIvVXNlcnMvZGFuaWVsL3NyYy9naXRodWIvRExSLVJNL3JsLWJhc2VsaW5lczMtem9vL3JsX3pvbzMvdXRpbHMucHmUjARmdW5jlIwdbGluZWFyX3NjaGVkdWxlLjxsb2NhbHM+LmZ1bmOUTTEBQxD4gADwDAAQIqBO0Q8y0AgylGgSjA5pbml0aWFsX3ZhbHVlX5SFlCl0lFKUfZQoaBiMB3JsX3pvbzOUaBqMDXJsX3pvbzMudXRpbHOUaByMQi9Vc2Vycy9kYW5pZWwvc3JjL2dpdGh1Yi9ETFItUk0vcmwtYmFzZWxpbmVzMy16b28vcmxfem9vMy91dGlscy5weZR1Tk5oHylSlIWUdJRSlGglaF19lH2UKGgajARmdW5jlGgpjB1saW5lYXJfc2NoZWR1bGUuPGxvY2Fscz4uZnVuY5RoK32UKIwScHJvZ3Jlc3NfcmVtYWluaW5nlIwIYnVpbHRpbnOUjAVmbG9hdJSTlIwGcmV0dXJulGhmdWgtTmguTmgvaFhoMGhLaDFoM0c/yZmZmZmZmoWUUpSFlIwXX2Nsb3VkcGlja2xlX3N1Ym1vZHVsZXOUXZSMC19fZ2xvYmFsc19flH2UdYaUhlIwhZRSlIWUaGtdlGhtfZR1hpSGUjCFlFKUhZRoa12UaG19lHWGlIZSMC4="
142
+ },
143
+ "clip_range_vf": null,
144
+ "normalize_advantage": true,
145
+ "target_kl": null,
146
+ "lr_schedule": {
147
+ ":type:": "<class 'function'>",
148
+ ":serialized:": "gAWV3wUAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLBUsTQyY+AZUAWwEAAAAAAAAAAFQBIgBVADUBAAAAAAAANQEAAAAAAAAkAJROhZSMBWZsb2F0lIWUjBJwcm9ncmVzc19yZW1haW5pbmeUhZSMgC9Vc2Vycy9kYW5pZWwvLmxvY2FsL3NoYXJlL3ZpcnR1YWxlbnZzL3JsLWJhc2VsaW5lczMtem9vLV83Wi01OF9vL2xpYi9weXRob24zLjEzL3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lIwIPGxhbWJkYT6UjCFnZXRfc2NoZWR1bGVfZm4uPGxvY2Fscz4uPGxhbWJkYT6US2JDEPiAAKRlqU7QO03TLE7UJk+UQwCUjA52YWx1ZV9zY2hlZHVsZZSFlCl0lFKUfZQojAtfX3BhY2thZ2VfX5SMGHN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbpSMCF9fbmFtZV9flIwec3RhYmxlX2Jhc2VsaW5lczMuY29tbW9uLnV0aWxzlIwIX19maWxlX1+UjIAvVXNlcnMvZGFuaWVsLy5sb2NhbC9zaGFyZS92aXJ0dWFsZW52cy9ybC1iYXNlbGluZXMzLXpvby1fN1otNThfby9saWIvcHl0aG9uMy4xMy9zaXRlLXBhY2thZ2VzL3N0YWJsZV9iYXNlbGluZXMzL2NvbW1vbi91dGlscy5weZR1Tk5oAIwQX21ha2VfZW1wdHlfY2VsbJSTlClSlIWUdJRSlGgAjBJfZnVuY3Rpb25fc2V0c3RhdGWUk5RoI32UfZQoaBqMCDxsYW1iZGE+lIwMX19xdWFsbmFtZV9flIwhZ2V0X3NjaGVkdWxlX2ZuLjxsb2NhbHM+LjxsYW1iZGE+lIwPX19hbm5vdGF0aW9uc19flH2UjA5fX2t3ZGVmYXVsdHNfX5ROjAxfX2RlZmF1bHRzX1+UTowKX19tb2R1bGVfX5RoG4wHX19kb2NfX5ROjAtfX2Nsb3N1cmVfX5RoAIwKX21ha2VfY2VsbJSTlGgCKGgHKEsBSwBLAEsBSwJLE0MOPgGVAFUAVAEtBQAAJACUjGQKUHJvZ3Jlc3Mgd2lsbCBkZWNyZWFzZSBmcm9tIDEgKGJlZ2lubmluZykgdG8gMAo6cGFyYW0gcHJvZ3Jlc3NfcmVtYWluaW5nOiAoZmxvYXQpCjpyZXR1cm46IChmbG9hdCkKlIWUKWgMhZSMQi9Vc2Vycy9kYW5pZWwvc3JjL2dpdGh1Yi9ETFItUk0vcmwtYmFzZWxpbmVzMy16b28vcmxfem9vMy91dGlscy5weZSMBGZ1bmOUjB1saW5lYXJfc2NoZWR1bGUuPGxvY2Fscz4uZnVuY5RNMQFDEPiAAPAMABAioE7RDzLQCDKUaBKMDmluaXRpYWxfdmFsdWVflIWUKXSUUpR9lChoGIwHcmxfem9vM5RoGowNcmxfem9vMy51dGlsc5RoHIxCL1VzZXJzL2RhbmllbC9zcmMvZ2l0aHViL0RMUi1STS9ybC1iYXNlbGluZXMzLXpvby9ybF96b28zL3V0aWxzLnB5lHVOTmgfKVKUhZR0lFKUaCVoR32UfZQoaBqMBGZ1bmOUaCmMHWxpbmVhcl9zY2hlZHVsZS48bG9jYWxzPi5mdW5jlGgrfZQojBJwcm9ncmVzc19yZW1haW5pbmeUjAhidWlsdGluc5SMBWZsb2F0lJOUjAZyZXR1cm6UaFB1aC1OaC5OaC9oQmgwaDVoMWgzRz9QYk3S8an8hZRSlIWUjBdfY2xvdWRwaWNrbGVfc3VibW9kdWxlc5RdlIwLX19nbG9iYWxzX1+UfZR1hpSGUjCFlFKUhZRoVV2UaFd9lHWGlIZSMC4="
149
+ }
150
+ }
ppo-CartPole-v1/policy.optimizer.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba832b3b7f519511e253f2db566257b488b66baf491abef244c27bd5421e88e7
3
+ size 82858
ppo-CartPole-v1/policy.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b5c93cb84a4c1845cca91bdcc2c2719664ba23e20fd7cbff5bf6c67ffcdcac9
3
+ size 41074
ppo-CartPole-v1/pytorch_variables.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebdad4b9cfe9cd22a3abadb5623bf7bb1f6eb2e408740245eb3f2044b0adc018
3
+ size 864
ppo-CartPole-v1/system_info.txt ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ - OS: macOS-14.7.4-arm64-arm-64bit-Mach-O Darwin Kernel Version 23.6.0: Thu Dec 19 20:44:10 PST 2024; root:xnu-10063.141.1.703.2~1/RELEASE_ARM64_T6000
2
+ - Python: 3.13.2
3
+ - Stable-Baselines3: 2.6.0a2
4
+ - PyTorch: 2.6.0
5
+ - GPU Enabled: False
6
+ - Numpy: 2.2.4
7
+ - Cloudpickle: 3.1.1
8
+ - Gymnasium: 1.1.1
9
+ - OpenAI Gym: 0.26.2
results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"mean_reward": 500.0, "std_reward": 0.0, "is_deterministic": true, "n_eval_episodes": 10, "eval_datetime": "2025-03-25T11:39:56.504292"}
train_eval_metrics.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:458f01b18f8f72eff92aba77905a43c43fb73b1c35453a505ad1971c10d010ac
3
+ size 9341