Commit
•
04924ec
1
Parent(s):
ad8b0f7
Initial commit
Browse files- .gitattributes +1 -0
- README.md +75 -0
- args.yml +83 -0
- config.yml +9 -0
- env_kwargs.yml +2 -0
- ppo-ReachCube-v0.zip +3 -0
- ppo-ReachCube-v0/_stable_baselines3_version +1 -0
- ppo-ReachCube-v0/data +99 -0
- ppo-ReachCube-v0/policy.optimizer.pth +3 -0
- ppo-ReachCube-v0/policy.pth +3 -0
- ppo-ReachCube-v0/pytorch_variables.pth +3 -0
- ppo-ReachCube-v0/system_info.txt +8 -0
- replay.mp4 +3 -0
- results.json +1 -0
- train_eval_metrics.zip +3 -0
.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,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: stable-baselines3
|
3 |
+
tags:
|
4 |
+
- ReachCube-v0
|
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: ReachCube-v0
|
16 |
+
type: ReachCube-v0
|
17 |
+
metrics:
|
18 |
+
- type: mean_reward
|
19 |
+
value: -88.73 +/- 25.37
|
20 |
+
name: mean_reward
|
21 |
+
verified: false
|
22 |
+
---
|
23 |
+
|
24 |
+
# **PPO** Agent playing **ReachCube-v0**
|
25 |
+
This is a trained model of a **PPO** agent playing **ReachCube-v0**
|
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 |
+
|
39 |
+
Install the RL Zoo (with SB3 and SB3-Contrib):
|
40 |
+
```bash
|
41 |
+
pip install rl_zoo3
|
42 |
+
```
|
43 |
+
|
44 |
+
```
|
45 |
+
# Download model and save it into the logs/ folder
|
46 |
+
python -m rl_zoo3.load_from_hub --algo ppo --env ReachCube-v0 -orga qgallouedec -f logs/
|
47 |
+
python -m rl_zoo3.enjoy --algo ppo --env ReachCube-v0 -f logs/
|
48 |
+
```
|
49 |
+
|
50 |
+
If you installed the RL Zoo3 via pip (`pip install rl_zoo3`), from anywhere you can do:
|
51 |
+
```
|
52 |
+
python -m rl_zoo3.load_from_hub --algo ppo --env ReachCube-v0 -orga qgallouedec -f logs/
|
53 |
+
python -m rl_zoo3.enjoy --algo ppo --env ReachCube-v0 -f logs/
|
54 |
+
```
|
55 |
+
|
56 |
+
## Training (with the RL Zoo)
|
57 |
+
```
|
58 |
+
python -m rl_zoo3.train --algo ppo --env ReachCube-v0 -f logs/
|
59 |
+
# Upload the model and generate video (when possible)
|
60 |
+
python -m rl_zoo3.push_to_hub --algo ppo --env ReachCube-v0 -f logs/ -orga qgallouedec
|
61 |
+
```
|
62 |
+
|
63 |
+
## Hyperparameters
|
64 |
+
```python
|
65 |
+
OrderedDict([('n_envs', 16),
|
66 |
+
('n_timesteps', 1000000.0),
|
67 |
+
('policy', 'MultiInputPolicy'),
|
68 |
+
('use_sde', True),
|
69 |
+
('normalize', False)])
|
70 |
+
```
|
71 |
+
|
72 |
+
# Environment Arguments
|
73 |
+
```python
|
74 |
+
{'observation_mode': 'state', 'render_mode': 'rgb_array'}
|
75 |
+
```
|
args.yml
ADDED
@@ -0,0 +1,83 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
!!python/object/apply:collections.OrderedDict
|
2 |
+
- - - algo
|
3 |
+
- ppo
|
4 |
+
- - conf_file
|
5 |
+
- null
|
6 |
+
- - device
|
7 |
+
- auto
|
8 |
+
- - env
|
9 |
+
- ReachCube-v0
|
10 |
+
- - env_kwargs
|
11 |
+
- observation_mode: state
|
12 |
+
- - eval_env_kwargs
|
13 |
+
- null
|
14 |
+
- - eval_episodes
|
15 |
+
- 5
|
16 |
+
- - eval_freq
|
17 |
+
- 25000
|
18 |
+
- - gym_packages
|
19 |
+
- - gym_lowcostrobot
|
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 |
+
- 1220238390
|
60 |
+
- - storage
|
61 |
+
- null
|
62 |
+
- - study_name
|
63 |
+
- null
|
64 |
+
- - tensorboard_log
|
65 |
+
- ''
|
66 |
+
- - track
|
67 |
+
- false
|
68 |
+
- - trained_agent
|
69 |
+
- ''
|
70 |
+
- - truncate_last_trajectory
|
71 |
+
- true
|
72 |
+
- - uuid
|
73 |
+
- false
|
74 |
+
- - vec_env
|
75 |
+
- dummy
|
76 |
+
- - verbose
|
77 |
+
- 1
|
78 |
+
- - wandb_entity
|
79 |
+
- null
|
80 |
+
- - wandb_project_name
|
81 |
+
- sb3
|
82 |
+
- - wandb_tags
|
83 |
+
- []
|
config.yml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
!!python/object/apply:collections.OrderedDict
|
2 |
+
- - - n_envs
|
3 |
+
- 16
|
4 |
+
- - n_timesteps
|
5 |
+
- 1000000.0
|
6 |
+
- - policy
|
7 |
+
- MultiInputPolicy
|
8 |
+
- - use_sde
|
9 |
+
- true
|
env_kwargs.yml
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
observation_mode: state
|
2 |
+
render_mode: rgb_array
|
ppo-ReachCube-v0.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a8b81204bb099d135c567f2839e5eaad30cd382f7fedd69a60d8b658ade93b4e
|
3 |
+
size 168995
|
ppo-ReachCube-v0/_stable_baselines3_version
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
2.3.2
|
ppo-ReachCube-v0/data
ADDED
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"policy_class": {
|
3 |
+
":type:": "<class 'abc.ABCMeta'>",
|
4 |
+
":serialized:": "gAWVRQAAAAAAAACMIXN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi5wb2xpY2llc5SMG011bHRpSW5wdXRBY3RvckNyaXRpY1BvbGljeZSTlC4=",
|
5 |
+
"__module__": "stable_baselines3.common.policies",
|
6 |
+
"__doc__": "\n MultiInputActorClass policy class for actor-critic algorithms (has both policy and value prediction).\n Used by A2C, PPO and the likes.\n\n :param observation_space: Observation space (Tuple)\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: Uses the CombinedExtractor\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 ",
|
7 |
+
"__init__": "<function MultiInputActorCriticPolicy.__init__ at 0x7efcb0798280>",
|
8 |
+
"__abstractmethods__": "frozenset()",
|
9 |
+
"_abc_impl": "<_abc._abc_data object at 0x7efcb0994f80>"
|
10 |
+
},
|
11 |
+
"verbose": 1,
|
12 |
+
"policy_kwargs": {},
|
13 |
+
"num_timesteps": 224928,
|
14 |
+
"_total_timesteps": 1000000,
|
15 |
+
"_num_timesteps_at_start": 0,
|
16 |
+
"seed": 0,
|
17 |
+
"action_noise": null,
|
18 |
+
"start_time": 1718288104583090574,
|
19 |
+
"learning_rate": 0.0003,
|
20 |
+
"tensorboard_log": null,
|
21 |
+
"_last_obs": null,
|
22 |
+
"_last_episode_starts": {
|
23 |
+
":type:": "<class 'numpy.ndarray'>",
|
24 |
+
":serialized:": "gAWVgwAAAAAAAACMEm51bXB5LmNvcmUubnVtZXJpY5SMC19mcm9tYnVmZmVylJOUKJYQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACUjAVudW1weZSMBWR0eXBllJOUjAJiMZSJiIeUUpQoSwOMAXyUTk5OSv////9K/////0sAdJRiSxCFlIwBQ5R0lFKULg=="
|
25 |
+
},
|
26 |
+
"_last_original_obs": null,
|
27 |
+
"_episode_num": 0,
|
28 |
+
"use_sde": true,
|
29 |
+
"sde_sample_freq": -1,
|
30 |
+
"_current_progress_remaining": 0.803392,
|
31 |
+
"_stats_window_size": 100,
|
32 |
+
"ep_info_buffer": {
|
33 |
+
":type:": "<class 'collections.deque'>",
|
34 |
+
":serialized:": "gAWVRAwAAAAAAACMC2NvbGxlY3Rpb25zlIwFZGVxdWWUk5QpS2SGlFKUKH2UKIwBcpRHwFohq814xDeMAWyUTfQBjAF0lEdAcFsNoJzDGnV9lChoBkfAWY9PnB+F12gHTfQBaAhHQHBZMjeKsMl1fZQoaAZHwFLZdt2s7uFoB030AWgIR0BwV1t1p0wKdX2UKGgGR8BdAry+Yc//aAdN9AFoCEdAcFT0tRNypHV9lChoBkfAYCXjMFEApGgHTfQBaAhHQHCNEhFEy+J1fZQoaAZHwF5h8xbjcVRoB030AWgIR0Bwiz71qWTpdX2UKGgGR8BWozvmYBvKaAdN9AFoCEdAcIlqn3ta6nV9lChoBkfAXZ1LnLaEjGgHTfQBaAhHQHCHmJm/WUd1fZQoaAZHwFV+sD4gzP9oB030AWgIR0BwhcfbKzRhdX2UKGgGR8BTsiMcZLqVaAdN9AFoCEdAcIP32mHgxnV9lChoBkfARhdVea8Yh2gHTfQBaAhHQHCCJLh73PB1fZQoaAZHwFpzvkBCD29oB030AWgIR0BwgFSDRMN+dX2UKGgGR8BPpWU8mrsCaAdN9AFoCEdAcH6DRc/t6XV9lChoBkfAV5tENOM2nGgHTfQBaAhHQHB8sRUWEbp1fZQoaAZHwFJQEehf0EpoB030AWgIR0Bwet6jWTX8dX2UKGgGR8BSXuEdvKlpaAdN9AFoCEdAcHjyZ8a4t3V9lChoBkfAWF3ZDiOvMmgHTfQBaAhHQHB3CZv1lGx1fZQoaAZHwGzVIGQjlgdoB030AWgIR0BwdS4NI9TxdX2UKGgGR8BRAGA9V3lkaAdN9AFoCEdAcHNXGOuJUHV9lChoBkfAXzVjriVB2WgHTfQBaAhHQHBw8C5mRNh1fZQoaAZHwEqh8ohIOH5oB030AWgIR0BwqSVeKKpDdX2UKGgGR8Bb8l/x2B8QaAdN9AFoCEdAcKdSS/0ulHV9lChoBkfAWPpRQ79ycWgHTfQBaAhHQHClfdl/Yrd1fZQoaAZHwFyGTH80k4ZoB030AWgIR0Bwo6vcJtzkdX2UKGgGR8BQtN4FA3UAaAdN9AFoCEdAcKHbGFSKnHV9lChoBkfATcywnpjc22gHTfQBaAhHQHCgCu+yquN1fZQoaAZHwFeCH6dlNDdoB030AWgIR0BwnjfLs8gZdX2UKGgGR8BPuI4+8oQWaAdN9AFoCEdAcJxnfVI7NnV9lChoBkfAVeH5FgDzRWgHTfQBaAhHQHCali4J/od1fZQoaAZHwFU8mOEM9bJoB030AWgIR0BwmMP5HmRvdX2UKGgGR8BdV2NR3u/laAdN9AFoCEdAcJbxgiNbT3V9lChoBkfAVbNHhCMP0GgHTfQBaAhHQHCVBT850bN1fZQoaAZHwFAH1Gb1AZ9oB030AWgIR0Bwkxx0dRzjdX2UKGgGR8Baf+GfwqiHaAdN9AFoCEdAcJFBAv+OwXV9lChoBkfAVG5ZTyauwGgHTfQBaAhHQHCPahL5AQh1fZQoaAZHwFv1hAnlXBBoB030AWgIR0BwjQMqjJuEdX2UKGgGR8BSVdTP0I1MaAdN9AFoCEdAc2P5I6KceHV9lChoBkfAUwetPpIMB2gHTfQBaAhHQHNiJxaPjn51fZQoaAZHwEpf9PUKArhoB030AWgIR0BzYFMtbs4UdX2UKGgGR8BRrCdBjWkKaAdN9AFoCEdAc16BbOeJ53V9lChoBkfAV+WG9Htnf2gHTfQBaAhHQHNcsPe54GF1fZQoaAZHwF2POp84PwxoB030AWgIR0BzWuFAVwgldX2UKGgGR8BSvhpYcNpeaAdN9AFoCEdAc1kOYIBzWHV9lChoBkfAVYE0Mw1zhmgHTfQBaAhHQHNXPmxMWXV1fZQoaAZHwEvc4PPLPldoB030AWgIR0BzVW3rleWwdX2UKGgGR8BXCJbMX7+DaAdN9AFoCEdAc1OclgMMJHV9lChoBkfAVUH6k690zWgHTfQBaAhHQHNRyvLX+VF1fZQoaAZHwFoH2NedCmdoB030AWgIR0BzT99c8kledX2UKGgGR8BkWjqMWGh3aAdN9AFoCEdAc033G4qgAnV9lChoBkfAWjl9JBgNPWgHTfQBaAhHQHNMG/ag2611fZQoaAZHwFdxYyfthNNoB030AWgIR0BzSkV8CxNZdX2UKGgGR8BWHtJrcj7iaAdN9AFoCEdAc0ffDDTBqXV9lChoBkfAUmK+49X9zmgHTfQBaAhHQHOCYvvjOs11fZQoaAZHwE7DiQ1aW5ZoB030AWgIR0BzgJAE+xGEdX2UKGgGR8BQgZEc81XOaAdN9AFoCEdAc367mdRR/HV9lChoBkfAVjbcVQAMlWgHTfQBaAhHQHN86aG5+Yt1fZQoaAZHwED8tf5ULlVoB030AWgIR0BzexkEs8PndX2UKGgGR8BbgTt9hJAdaAdN9AFoCEdAc3lI42jwhHV9lChoBkfATYT1M/QjU2gHTfQBaAhHQHN3dbxEv011fZQoaAZHwESeRbr1M/RoB030AWgIR0BzdaV6eGwidX2UKGgGR8BOKjQZ4wAVaAdN9AFoCEdAc3PUMoc7yXV9lChoBkfASnuvW6K+BmgHTfQBaAhHQHNyAgTyrgh1fZQoaAZHwFS7WJJoTPBoB030AWgIR0BzcC+IuXeFdX2UKGgGR8BUrnerMkhSaAdN9AFoCEdAc25DTjNpunV9lChoBkfARpTFMqSX+mgHTfQBaAhHQHNsWn4wh4d1fZQoaAZHwE3mUX531SRoB030AWgIR0Bzan7uUliSdX2UKGgGR8Bnf56KLsKLaAdN9AFoCEdAc2in+AEt/XV9lChoBkfAUFnYWcjJMmgHTfQBaAhHQHNmQQ6IWP91fZQoaAZHwFxV0HyEtd1oB030AWgIR0BzoR5OafBfdX2UKGgGR8BY3KSowVTKaAdN9AFoCEdAc59MdcSoO3V9lChoBkfAVd5/0/W1+mgHTfQBaAhHQHOdeHN5dGB1fZQoaAZHwF6tlPrOZ9doB030AWgIR0Bzm6bKA8SxdX2UKGgGR8BTkVxS5y2haAdN9AFoCEdAc5nWYnfEXXV9lChoBkfAWlvqmj0tiGgHTfQBaAhHQHOYBoqTbFl1fZQoaAZHwFpVp+MIeHVoB030AWgIR0BzljO6d1+zdX2UKGgGR8BG3zf779AHaAdN9AFoCEdAc5Rj1wo9cXV9lChoBkfAVfndsSCe3GgHTfQBaAhHQHOSkt/WlM11fZQoaAZHwFwSzlLeyiVoB030AWgIR0BzkMEJSiuddX2UKGgGR8BUh/VNHpbEaAdN9AFoCEdAc47u4wyqMnV9lChoBkfAUdBL5AQg92gHTfQBaAhHQHONAvpQk5Z1fZQoaAZHwFoAsDGLk0doB030AWgIR0BzixqBVdX1dX2UKGgGR8Bdon9aUzKtaAdN9AFoCEdAc4k/O+qR2nV9lChoBkfAT+2cYqG1yGgHTfQBaAhHQHOHaKtPpIN1fZQoaAZHwFb+fQ8fV7RoB030AWgIR0BzhQIPbwjMdX2UKGgGR8BXtRMrVe8gaAdN9AFoCEdAc7+z+FUQ1HV9lChoBkfAWu32oNutOmgHTfQBaAhHQHO94RZlnRN1fZQoaAZHwE3AiTMaCMBoB030AWgIR0BzvAzyjHn2dX2UKGgGR8BNqtedCmdiaAdN9AFoCEdAc7o7A+IM0HV9lChoBkfARKnFglWwNmgHTfQBaAhHQHO4ak/KQq91fZQoaAZHwFO5Ljghr31oB030AWgIR0Bztpo8IRh+dX2UKGgGR8BBrE1EVnEmaAdN9AFoCEdAc7THIIWxhXV9lChoBkfARDMMEzO5a2gHTfQBaAhHQHOy9vCMxXZ1fZQoaAZHwF0beaa1Cw9oB030AWgIR0BzsSXAuZkTdX2UKGgGR8BQPLiEQGwBaAdN9AFoCEdAc69TodMj/3V9lChoBkfAWS893bEgn2gHTfQBaAhHQHOtgTmGM4t1fZQoaAZHwFd+ylenhsJoB030AWgIR0Bzq5UNrj5sdX2UKGgGR8BXb2WdEsreaAdN9AFoCEdAc6msS00FbHV9lChoBkfAWA0pRXOnmGgHTfQBaAhHQHOn0Nz8xbl1fZQoaAZHwEKuT+NtIkJoB030AWgIR0Bzpfn0TURWdX2UKGgGR8BX29eyAxzraAdN9AFoCEdAc6OTGYKIBXVlLg=="
|
35 |
+
},
|
36 |
+
"ep_success_buffer": {
|
37 |
+
":type:": "<class 'collections.deque'>",
|
38 |
+
":serialized:": "gAWVIAAAAAAAAACMC2NvbGxlY3Rpb25zlIwFZGVxdWWUk5QpS2SGlFKULg=="
|
39 |
+
},
|
40 |
+
"_n_updates": 60,
|
41 |
+
"observation_space": {
|
42 |
+
":type:": "<class 'gymnasium.spaces.dict.Dict'>",
|
43 |
+
":serialized:": "gAWVzAMAAAAAAACMFWd5bW5hc2l1bS5zcGFjZXMuZGljdJSMBERpY3SUk5QpgZR9lCiMBnNwYWNlc5SMC2NvbGxlY3Rpb25zlIwLT3JkZXJlZERpY3SUk5QpUpQojAhhcm1fcXBvc5SMFGd5bW5hc2l1bS5zcGFjZXMuYm94lIwDQm94lJOUKYGUfZQojAVkdHlwZZSMBW51bXB5lIwFZHR5cGWUk5SMAmY0lImIh5RSlChLA4wBPJROTk5K/////0r/////SwB0lGKMDWJvdW5kZWRfYmVsb3eUjBJudW1weS5jb3JlLm51bWVyaWOUjAtfZnJvbWJ1ZmZlcpSTlCiWBgAAAAAAAAABAQEBAQGUaBOMAmIxlImIh5RSlChLA4wBfJROTk5K/////0r/////SwB0lGJLBoWUjAFDlHSUUpSMDWJvdW5kZWRfYWJvdmWUaBwolgYAAAAAAAAAAQEBAQEBlGggSwaFlGgkdJRSlIwGX3NoYXBllEsGhZSMA2xvd5RoHCiWGAAAAAAAAADbD0nA2w9JwNsPScDbD0nA2w9JwNsPScCUaBZLBoWUaCR0lFKUjARoaWdolGgcKJYYAAAAAAAAANsPSUDbD0lA2w9JQNsPSUDbD0lA2w9JQJRoFksGhZRoJHSUUpSMCGxvd19yZXBylIwKLTMuMTQxNTkyN5SMCWhpZ2hfcmVwcpSMCTMuMTQxNTkyN5SMCl9ucF9yYW5kb22UTnVijAhhcm1fcXZlbJRoDSmBlH2UKGgQaBZoGWgcKJYGAAAAAAAAAAEBAQEBAZRoIEsGhZRoJHSUUpRoJ2gcKJYGAAAAAAAAAAEBAQEBAZRoIEsGhZRoJHSUUpRoLEsGhZRoLmgcKJYYAAAAAAAAAAAAIMEAACDBAAAgwQAAIMEAACDBAAAgwZRoFksGhZRoJHSUUpRoM2gcKJYYAAAAAAAAAAAAIEEAACBBAAAgQQAAIEEAACBBAAAgQZRoFksGhZRoJHSUUpRoOIwFLTEwLjCUaDqMBDEwLjCUaDxOdWKMCGN1YmVfcG9zlGgNKYGUfZQoaBBoFmgZaBwolgMAAAAAAAAAAQEBlGggSwOFlGgkdJRSlGgnaBwolgMAAAAAAAAAAQEBlGggSwOFlGgkdJRSlGgsSwOFlGguaBwolgwAAAAAAAAAAAAgwQAAIMEAACDBlGgWSwOFlGgkdJRSlGgzaBwolgwAAAAAAAAAAAAgQQAAIEEAACBBlGgWSwOFlGgkdJRSlGg4jAUtMTAuMJRoOowEMTAuMJRoPE51YnVoLE5oEE5oPE51Yi4=",
|
44 |
+
"spaces": "OrderedDict([('arm_qpos', Box(-3.1415927, 3.1415927, (6,), float32)), ('arm_qvel', Box(-10.0, 10.0, (6,), float32)), ('cube_pos', Box(-10.0, 10.0, (3,), float32))])",
|
45 |
+
"_shape": null,
|
46 |
+
"dtype": null,
|
47 |
+
"_np_random": null
|
48 |
+
},
|
49 |
+
"action_space": {
|
50 |
+
":type:": "<class 'gymnasium.spaces.box.Box'>",
|
51 |
+
":serialized:": "gAWVfgIAAAAAAACMFGd5bW5hc2l1bS5zcGFjZXMuYm94lIwDQm94lJOUKYGUfZQojAVkdHlwZZSMBW51bXB5lIwFZHR5cGWUk5SMAmY0lImIh5RSlChLA4wBPJROTk5K/////0r/////SwB0lGKMDWJvdW5kZWRfYmVsb3eUjBJudW1weS5jb3JlLm51bWVyaWOUjAtfZnJvbWJ1ZmZlcpSTlCiWBgAAAAAAAAABAQEBAQGUaAiMAmIxlImIh5RSlChLA4wBfJROTk5K/////0r/////SwB0lGJLBoWUjAFDlHSUUpSMDWJvdW5kZWRfYWJvdmWUaBEolgYAAAAAAAAAAQEBAQEBlGgVSwaFlGgZdJRSlIwGX3NoYXBllEsGhZSMA2xvd5RoESiWGAAAAAAAAAAAAIC/AACAvwAAgL8AAIC/AACAvwAAgL+UaAtLBoWUaBl0lFKUjARoaWdolGgRKJYYAAAAAAAAAAAAgD8AAIA/AACAPwAAgD8AAIA/AACAP5RoC0sGhZRoGXSUUpSMCGxvd19yZXBylIwELTEuMJSMCWhpZ2hfcmVwcpSMAzEuMJSMCl9ucF9yYW5kb22UjBRudW1weS5yYW5kb20uX3BpY2tsZZSMEF9fZ2VuZXJhdG9yX2N0b3KUk5SMBVBDRzY0lGgyjBRfX2JpdF9nZW5lcmF0b3JfY3RvcpSTlIaUUpR9lCiMDWJpdF9nZW5lcmF0b3KUjAVQQ0c2NJSMBXN0YXRllH2UKGg9ihDjYZWmt15YCS1Fllk0taEajANpbmOUihCpc3hEvDOBWIIa9zrb2o1BdYwKaGFzX3VpbnQzMpRLAIwIdWludGVnZXKUSwB1YnViLg==",
|
52 |
+
"dtype": "float32",
|
53 |
+
"bounded_below": "[ True True True True True True]",
|
54 |
+
"bounded_above": "[ True True True True True True]",
|
55 |
+
"_shape": [
|
56 |
+
6
|
57 |
+
],
|
58 |
+
"low": "[-1. -1. -1. -1. -1. -1.]",
|
59 |
+
"high": "[1. 1. 1. 1. 1. 1.]",
|
60 |
+
"low_repr": "-1.0",
|
61 |
+
"high_repr": "1.0",
|
62 |
+
"_np_random": "Generator(PCG64)"
|
63 |
+
},
|
64 |
+
"n_envs": 1,
|
65 |
+
"n_steps": 2048,
|
66 |
+
"gamma": 0.99,
|
67 |
+
"gae_lambda": 0.95,
|
68 |
+
"ent_coef": 0.0,
|
69 |
+
"vf_coef": 0.5,
|
70 |
+
"max_grad_norm": 0.5,
|
71 |
+
"rollout_buffer_class": {
|
72 |
+
":type:": "<class 'abc.ABCMeta'>",
|
73 |
+
":serialized:": "gAWVOgAAAAAAAACMIHN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi5idWZmZXJzlIwRRGljdFJvbGxvdXRCdWZmZXKUk5Qu",
|
74 |
+
"__module__": "stable_baselines3.common.buffers",
|
75 |
+
"__annotations__": "{'observation_space': <class 'gymnasium.spaces.dict.Dict'>, 'obs_shape': typing.Dict[str, typing.Tuple[int, ...]], 'observations': typing.Dict[str, numpy.ndarray]}",
|
76 |
+
"__doc__": "\n Dict Rollout buffer used in on-policy algorithms like A2C/PPO.\n Extends the RolloutBuffer to use dictionary observations\n\n It corresponds to ``buffer_size`` transitions collected\n using the current policy.\n This experience will be discarded after the policy update.\n In order to use PPO objective, we also store the current value of each state\n and the log probability of each taken action.\n\n The term rollout here refers to the model-free notion and should not\n be used with the concept of rollout used in model-based RL or planning.\n Hence, 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 Monte-Carlo advantage estimate when set to 1.\n :param gamma: Discount factor\n :param n_envs: Number of parallel environments\n ",
|
77 |
+
"__init__": "<function DictRolloutBuffer.__init__ at 0x7efcb0aaa5f0>",
|
78 |
+
"reset": "<function DictRolloutBuffer.reset at 0x7efcb0aaa680>",
|
79 |
+
"add": "<function DictRolloutBuffer.add at 0x7efcb0aaa710>",
|
80 |
+
"get": "<function DictRolloutBuffer.get at 0x7efcb0aaa7a0>",
|
81 |
+
"_get_samples": "<function DictRolloutBuffer._get_samples at 0x7efcb0aaa830>",
|
82 |
+
"__abstractmethods__": "frozenset()",
|
83 |
+
"_abc_impl": "<_abc._abc_data object at 0x7efcb0aa4900>"
|
84 |
+
},
|
85 |
+
"rollout_buffer_kwargs": {},
|
86 |
+
"batch_size": 64,
|
87 |
+
"n_epochs": 10,
|
88 |
+
"clip_range": {
|
89 |
+
":type:": "<class 'function'>",
|
90 |
+
":serialized:": "gAWVrQUAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLA0sTQwx0AIgAfACDAYMBUwCUToWUjAVmbG9hdJSFlIwScHJvZ3Jlc3NfcmVtYWluaW5nlIWUjHAvZnN4L3FnYWxsb3VlZGVjL21pbmljb25kYTMvZW52cy9neW1fbG93Y29zdHJvYm90L2xpYi9weXRob24zLjEwL3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lIwIPGxhbWJkYT6US2FDAgwAlIwOdmFsdWVfc2NoZWR1bGWUhZQpdJRSlH2UKIwLX19wYWNrYWdlX1+UjBhzdGFibGVfYmFzZWxpbmVzMy5jb21tb26UjAhfX25hbWVfX5SMHnN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi51dGlsc5SMCF9fZmlsZV9flIxwL2ZzeC9xZ2FsbG91ZWRlYy9taW5pY29uZGEzL2VudnMvZ3ltX2xvd2Nvc3Ryb2JvdC9saWIvcHl0aG9uMy4xMC9zaXRlLXBhY2thZ2VzL3N0YWJsZV9iYXNlbGluZXMzL2NvbW1vbi91dGlscy5weZR1Tk5oAIwQX21ha2VfZW1wdHlfY2VsbJSTlClSlIWUdJRSlGgAjBJfZnVuY3Rpb25fc2V0c3RhdGWUk5RoIX2UfZQoaBhoD4wMX19xdWFsbmFtZV9flIwhZ2V0X3NjaGVkdWxlX2ZuLjxsb2NhbHM+LjxsYW1iZGE+lIwPX19hbm5vdGF0aW9uc19flH2UjA5fX2t3ZGVmYXVsdHNfX5ROjAxfX2RlZmF1bHRzX1+UTowKX19tb2R1bGVfX5RoGYwHX19kb2NfX5ROjAtfX2Nsb3N1cmVfX5RoAIwKX21ha2VfY2VsbJSTlGgCKGgHKEsBSwBLAEsBSwNLE0MMdACIAHwAgwGDAVMAlE6FlGgKhZRoDIWUjHAvZnN4L3FnYWxsb3VlZGVjL21pbmljb25kYTMvZW52cy9neW1fbG93Y29zdHJvYm90L2xpYi9weXRob24zLjEwL3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lIwIPGxhbWJkYT6US2FDAgwAlGgRhZQpdJRSlH2UKGgWjBhzdGFibGVfYmFzZWxpbmVzMy5jb21tb26UaBiMHnN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi51dGlsc5RoGoxwL2ZzeC9xZ2FsbG91ZWRlYy9taW5pY29uZGEzL2VudnMvZ3ltX2xvd2Nvc3Ryb2JvdC9saWIvcHl0aG9uMy4xMC9zaXRlLXBhY2thZ2VzL3N0YWJsZV9iYXNlbGluZXMzL2NvbW1vbi91dGlscy5weZR1Tk5oHSlSlIWUdJRSlGgjaEJ9lH2UKGgYaDZoJowhZ2V0X3NjaGVkdWxlX2ZuLjxsb2NhbHM+LjxsYW1iZGE+lGgofZRoKk5oK05oLGg9aC1OaC5oMGgCKGgHKEsBSwBLAEsBSwFLE0MEiABTAJRoMimMAV+UhZRoNYwEZnVuY5RLhUMCBAGUjAN2YWyUhZQpdJRSlGg7Tk5oHSlSlIWUdJRSlGgjaFN9lH2UKGgYaEpoJowZY29uc3RhbnRfZm4uPGxvY2Fscz4uZnVuY5RoKH2UaCpOaCtOaCxoPWgtTmguaDBHP8mZmZmZmZqFlFKUhZSMF19jbG91ZHBpY2tsZV9zdWJtb2R1bGVzlF2UjAtfX2dsb2JhbHNfX5R9lHWGlIZSMIWUUpSFlGhbXZRoXX2UdYaUhlIwhZRSlIWUaFtdlGhdfZR1hpSGUjAu"
|
91 |
+
},
|
92 |
+
"clip_range_vf": null,
|
93 |
+
"normalize_advantage": true,
|
94 |
+
"target_kl": null,
|
95 |
+
"lr_schedule": {
|
96 |
+
":type:": "<class 'function'>",
|
97 |
+
":serialized:": "gAWV0QMAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLA0sTQwx0AIgAfACDAYMBUwCUToWUjAVmbG9hdJSFlIwScHJvZ3Jlc3NfcmVtYWluaW5nlIWUjHAvZnN4L3FnYWxsb3VlZGVjL21pbmljb25kYTMvZW52cy9neW1fbG93Y29zdHJvYm90L2xpYi9weXRob24zLjEwL3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lIwIPGxhbWJkYT6US2FDAgwAlIwOdmFsdWVfc2NoZWR1bGWUhZQpdJRSlH2UKIwLX19wYWNrYWdlX1+UjBhzdGFibGVfYmFzZWxpbmVzMy5jb21tb26UjAhfX25hbWVfX5SMHnN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi51dGlsc5SMCF9fZmlsZV9flIxwL2ZzeC9xZ2FsbG91ZWRlYy9taW5pY29uZGEzL2VudnMvZ3ltX2xvd2Nvc3Ryb2JvdC9saWIvcHl0aG9uMy4xMC9zaXRlLXBhY2thZ2VzL3N0YWJsZV9iYXNlbGluZXMzL2NvbW1vbi91dGlscy5weZR1Tk5oAIwQX21ha2VfZW1wdHlfY2VsbJSTlClSlIWUdJRSlGgAjBJfZnVuY3Rpb25fc2V0c3RhdGWUk5RoIX2UfZQoaBhoD4wMX19xdWFsbmFtZV9flIwhZ2V0X3NjaGVkdWxlX2ZuLjxsb2NhbHM+LjxsYW1iZGE+lIwPX19hbm5vdGF0aW9uc19flH2UjA5fX2t3ZGVmYXVsdHNfX5ROjAxfX2RlZmF1bHRzX1+UTowKX19tb2R1bGVfX5RoGYwHX19kb2NfX5ROjAtfX2Nsb3N1cmVfX5RoAIwKX21ha2VfY2VsbJSTlGgCKGgHKEsBSwBLAEsBSwFLE0MEiABTAJRoCSmMAV+UhZRoDowEZnVuY5RLhUMCBAGUjAN2YWyUhZQpdJRSlGgVTk5oHSlSlIWUdJRSlGgjaD19lH2UKGgYaDRoJowZY29uc3RhbnRfZm4uPGxvY2Fscz4uZnVuY5RoKH2UaCpOaCtOaCxoGWgtTmguaDBHPzOpKjBVMmGFlFKUhZSMF19jbG91ZHBpY2tsZV9zdWJtb2R1bGVzlF2UjAtfX2dsb2JhbHNfX5R9lHWGlIZSMIWUUpSFlGhFXZRoR32UdYaUhlIwLg=="
|
98 |
+
}
|
99 |
+
}
|
ppo-ReachCube-v0/policy.optimizer.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b7d5261c33e949ce9653db1467405f9e8b42aefbe93b99dab567d723a3e99c76
|
3 |
+
size 100513
|
ppo-ReachCube-v0/policy.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a253ba7489cd44561c32dfbd274d7a08449dd06a1141ea495120b5986f677a3a
|
3 |
+
size 50159
|
ppo-ReachCube-v0/pytorch_variables.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0c35cea3b2e60fb5e7e162d3592df775cd400e575a31c72f359fb9e654ab00c5
|
3 |
+
size 864
|
ppo-ReachCube-v0/system_info.txt
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
- OS: Linux-5.15.0-1048-aws-x86_64-with-glibc2.31 # 53~20.04.1-Ubuntu SMP Wed Oct 4 16:44:20 UTC 2023
|
2 |
+
- Python: 3.10.14
|
3 |
+
- Stable-Baselines3: 2.3.2
|
4 |
+
- PyTorch: 2.3.1+cu121
|
5 |
+
- GPU Enabled: True
|
6 |
+
- Numpy: 1.26.4
|
7 |
+
- Cloudpickle: 3.0.0
|
8 |
+
- Gymnasium: 0.29.1
|
replay.mp4
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ef591af56c4bb08d448284be50c95737a86184b90e8dac446a023ad565e21175
|
3 |
+
size 658863
|
results.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"mean_reward": -88.7257652, "std_reward": 25.37057335271054, "is_deterministic": true, "n_eval_episodes": 10, "eval_datetime": "2024-06-13T14:21:41.684416"}
|
train_eval_metrics.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d13b306e66c99b28b65658a5d346091237f3e4988aff908fb3575dad1b9a013e
|
3 |
+
size 18156
|