5m steps
Browse files- .gitattributes +1 -0
- Heli.onnx +3 -0
- Heli/Heli-1249792.onnx +3 -0
- Heli/Heli-1249792.pt +3 -0
- Heli/Heli-1398016.onnx +3 -0
- Heli/Heli-1398016.pt +3 -0
- Heli/Heli-249993.onnx +3 -0
- Heli/Heli-249993.pt +3 -0
- Heli/Heli-499984.onnx +3 -0
- Heli/Heli-499984.pt +3 -0
- Heli/Heli-749884.onnx +3 -0
- Heli/Heli-749884.pt +3 -0
- Heli/Heli-999718.onnx +3 -0
- Heli/Heli-999718.pt +3 -0
- Heli/checkpoint.pt +3 -0
- Heli/events.out.tfevents.1691441418.28cc7e6db715.1368.0 +3 -0
- README.md +35 -0
- config.json +1 -0
- configuration.yaml +147 -0
- run_logs/Player-0.log +3 -0
- run_logs/timers.json +180 -0
- run_logs/training_status.json +74 -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 |
+
run_logs/Player-0.log filter=lfs diff=lfs merge=lfs -text
|
Heli.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:efafe4fde91a35216e645d3efff889a6140087bb3d17806008345e6c9fc08710
|
3 |
+
size 2187053
|
Heli/Heli-1249792.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5560542367f4d30034fceb58188298bf94bf8b5440f126ed815a8432af21bbd8
|
3 |
+
size 2187053
|
Heli/Heli-1249792.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f0490f426509bc1c31a72d9753a52fe22d304d482501d9b247fde4ac0f0a4ab8
|
3 |
+
size 14922498
|
Heli/Heli-1398016.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:efafe4fde91a35216e645d3efff889a6140087bb3d17806008345e6c9fc08710
|
3 |
+
size 2187053
|
Heli/Heli-1398016.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d70f328425d5fd9f32505dec64cc177dde25cd54afd7b5dd1681265b8962a318
|
3 |
+
size 14922498
|
Heli/Heli-249993.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e6d60ad210e3e1a409ff986e4b3ad41e825878043457abf2ac2e8b7d42d84f9b
|
3 |
+
size 2187053
|
Heli/Heli-249993.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84ae0b59909825197bfb9f1d01a419fce7a08c627bdaa0d7ae1bde2f49bcc439
|
3 |
+
size 14922498
|
Heli/Heli-499984.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:be30046a600ab887228ccdbb9bc19e7fc985349762f1f2442d0bcc575ec49f4e
|
3 |
+
size 2187053
|
Heli/Heli-499984.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f93f9e40157a743d7c612ac8e540b4f1ffeb7b5c53a1a5e2624622b8c154b99d
|
3 |
+
size 14922498
|
Heli/Heli-749884.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7bc232165b2b06c191e692988ee9fd8ba5fcf1e9a5bb0b8219e845cf24d90bd3
|
3 |
+
size 2187053
|
Heli/Heli-749884.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e54e601f66908a3426241ca1f63ccde6f6672b52da67c15d9c3f0006cd507828
|
3 |
+
size 14922498
|
Heli/Heli-999718.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:20bb392c627e57bcd0ce299d89676b47149d6249678aa44aa5415985d6068284
|
3 |
+
size 2187053
|
Heli/Heli-999718.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7795e9702140a59ca4c386bdef0af0582a7862eaf02ab60527f1f49b5e7ddf2d
|
3 |
+
size 14922498
|
Heli/checkpoint.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d70f328425d5fd9f32505dec64cc177dde25cd54afd7b5dd1681265b8962a318
|
3 |
+
size 14922498
|
Heli/events.out.tfevents.1691441418.28cc7e6db715.1368.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23d97b12129a198b7a439b5ea23cb4646623ceb9ab99124021c111cfebd6b20a
|
3 |
+
size 529124
|
README.md
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: ml-agents
|
3 |
+
tags:
|
4 |
+
- Heli
|
5 |
+
- deep-reinforcement-learning
|
6 |
+
- reinforcement-learning
|
7 |
+
- ML-Agents-Heli
|
8 |
+
---
|
9 |
+
|
10 |
+
# **ppo** Agent playing **Heli**
|
11 |
+
This is a trained model of a **ppo** agent playing **Heli**
|
12 |
+
using the [Unity ML-Agents Library](https://github.com/Unity-Technologies/ml-agents).
|
13 |
+
|
14 |
+
## Usage (with ML-Agents)
|
15 |
+
The Documentation: https://unity-technologies.github.io/ml-agents/ML-Agents-Toolkit-Documentation/
|
16 |
+
|
17 |
+
We wrote a complete tutorial to learn to train your first agent using ML-Agents and publish it to the Hub:
|
18 |
+
- A *short tutorial* where you teach Huggy the Dog 🐶 to fetch the stick and then play with him directly in your
|
19 |
+
browser: https://huggingface.co/learn/deep-rl-course/unitbonus1/introduction
|
20 |
+
- A *longer tutorial* to understand how works ML-Agents:
|
21 |
+
https://huggingface.co/learn/deep-rl-course/unit5/introduction
|
22 |
+
|
23 |
+
### Resume the training
|
24 |
+
```bash
|
25 |
+
mlagents-learn <your_configuration_file_path.yaml> --run-id=<run_id> --resume
|
26 |
+
```
|
27 |
+
|
28 |
+
### Watch your Agent play
|
29 |
+
You can watch your agent **playing directly in your browser**
|
30 |
+
|
31 |
+
1. If the environment is part of ML-Agents official environments, go to https://huggingface.co/unity
|
32 |
+
2. Step 1: Find your model_id: AtilliO/Chopper03_00
|
33 |
+
3. Step 2: Select your *.nn /*.onnx file
|
34 |
+
4. Click on Watch the agent play 👀
|
35 |
+
|
config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"default_settings": null, "behaviors": {"Heli": {"trainer_type": "ppo", "hyperparameters": {"batch_size": 256, "buffer_size": 10240, "learning_rate": 0.0002, "beta": 0.001, "epsilon": 0.2, "lambd": 0.95, "num_epoch": 3, "shared_critic": false, "learning_rate_schedule": "linear", "beta_schedule": "linear", "epsilon_schedule": "linear"}, "checkpoint_interval": 250000, "network_settings": {"normalize": true, "hidden_units": 512, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "reward_signals": {"extrinsic": {"gamma": 0.99, "strength": 1.0, "network_settings": {"normalize": false, "hidden_units": 128, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}}, "curiosity": {"gamma": 0.99, "strength": 0.02, "network_settings": {"normalize": false, "hidden_units": 256, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "learning_rate": 0.0003, "encoding_size": null}}, "init_path": null, "keep_checkpoints": 15, "even_checkpoints": false, "max_steps": 5000000, "time_horizon": 512, "summary_freq": 5000, "threaded": true, "self_play": null, "behavioral_cloning": null}, "LoaderG": {"trainer_type": "ppo", "hyperparameters": {"batch_size": 128, "buffer_size": 2048, "learning_rate": 0.0003, "beta": 0.01, "epsilon": 0.2, "lambd": 0.95, "num_epoch": 3, "shared_critic": false, "learning_rate_schedule": "linear", "beta_schedule": "linear", "epsilon_schedule": "linear"}, "checkpoint_interval": 500000, "network_settings": {"normalize": false, "hidden_units": 512, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "reward_signals": {"extrinsic": {"gamma": 0.99, "strength": 1.0, "network_settings": {"normalize": false, "hidden_units": 128, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}}, "curiosity": {"gamma": 0.99, "strength": 0.02, "network_settings": {"normalize": false, "hidden_units": 256, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "learning_rate": 0.0003, "encoding_size": null}}, "init_path": null, "keep_checkpoints": 5, "even_checkpoints": false, "max_steps": 2000000, "time_horizon": 512, "summary_freq": 5000, "threaded": true, "self_play": null, "behavioral_cloning": null}}, "env_settings": {"env_path": "./training-envs-executables/linux/ChopperLnx03/ChopperLnx03", "env_args": null, "base_port": 5005, "num_envs": 1, "num_areas": 1, "seed": -1, "max_lifetime_restarts": 10, "restarts_rate_limit_n": 1, "restarts_rate_limit_period_s": 60}, "engine_settings": {"width": 84, "height": 84, "quality_level": 5, "time_scale": 20, "target_frame_rate": -1, "capture_frame_rate": 60, "no_graphics": true}, "environment_parameters": null, "checkpoint_settings": {"run_id": "Chopper03_00", "initialize_from": null, "load_model": false, "resume": false, "force": false, "train_model": false, "inference": false, "results_dir": "results"}, "torch_settings": {"device": null}, "debug": false}
|
configuration.yaml
ADDED
@@ -0,0 +1,147 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
default_settings: null
|
2 |
+
behaviors:
|
3 |
+
Heli:
|
4 |
+
trainer_type: ppo
|
5 |
+
hyperparameters:
|
6 |
+
batch_size: 256
|
7 |
+
buffer_size: 10240
|
8 |
+
learning_rate: 0.0002
|
9 |
+
beta: 0.001
|
10 |
+
epsilon: 0.2
|
11 |
+
lambd: 0.95
|
12 |
+
num_epoch: 3
|
13 |
+
shared_critic: false
|
14 |
+
learning_rate_schedule: linear
|
15 |
+
beta_schedule: linear
|
16 |
+
epsilon_schedule: linear
|
17 |
+
checkpoint_interval: 250000
|
18 |
+
network_settings:
|
19 |
+
normalize: true
|
20 |
+
hidden_units: 512
|
21 |
+
num_layers: 2
|
22 |
+
vis_encode_type: simple
|
23 |
+
memory: null
|
24 |
+
goal_conditioning_type: hyper
|
25 |
+
deterministic: false
|
26 |
+
reward_signals:
|
27 |
+
extrinsic:
|
28 |
+
gamma: 0.99
|
29 |
+
strength: 1.0
|
30 |
+
network_settings:
|
31 |
+
normalize: false
|
32 |
+
hidden_units: 128
|
33 |
+
num_layers: 2
|
34 |
+
vis_encode_type: simple
|
35 |
+
memory: null
|
36 |
+
goal_conditioning_type: hyper
|
37 |
+
deterministic: false
|
38 |
+
curiosity:
|
39 |
+
gamma: 0.99
|
40 |
+
strength: 0.02
|
41 |
+
network_settings:
|
42 |
+
normalize: false
|
43 |
+
hidden_units: 256
|
44 |
+
num_layers: 2
|
45 |
+
vis_encode_type: simple
|
46 |
+
memory: null
|
47 |
+
goal_conditioning_type: hyper
|
48 |
+
deterministic: false
|
49 |
+
learning_rate: 0.0003
|
50 |
+
encoding_size: null
|
51 |
+
init_path: null
|
52 |
+
keep_checkpoints: 15
|
53 |
+
even_checkpoints: false
|
54 |
+
max_steps: 5000000
|
55 |
+
time_horizon: 512
|
56 |
+
summary_freq: 5000
|
57 |
+
threaded: true
|
58 |
+
self_play: null
|
59 |
+
behavioral_cloning: null
|
60 |
+
LoaderG:
|
61 |
+
trainer_type: ppo
|
62 |
+
hyperparameters:
|
63 |
+
batch_size: 128
|
64 |
+
buffer_size: 2048
|
65 |
+
learning_rate: 0.0003
|
66 |
+
beta: 0.01
|
67 |
+
epsilon: 0.2
|
68 |
+
lambd: 0.95
|
69 |
+
num_epoch: 3
|
70 |
+
shared_critic: false
|
71 |
+
learning_rate_schedule: linear
|
72 |
+
beta_schedule: linear
|
73 |
+
epsilon_schedule: linear
|
74 |
+
checkpoint_interval: 500000
|
75 |
+
network_settings:
|
76 |
+
normalize: false
|
77 |
+
hidden_units: 512
|
78 |
+
num_layers: 2
|
79 |
+
vis_encode_type: simple
|
80 |
+
memory: null
|
81 |
+
goal_conditioning_type: hyper
|
82 |
+
deterministic: false
|
83 |
+
reward_signals:
|
84 |
+
extrinsic:
|
85 |
+
gamma: 0.99
|
86 |
+
strength: 1.0
|
87 |
+
network_settings:
|
88 |
+
normalize: false
|
89 |
+
hidden_units: 128
|
90 |
+
num_layers: 2
|
91 |
+
vis_encode_type: simple
|
92 |
+
memory: null
|
93 |
+
goal_conditioning_type: hyper
|
94 |
+
deterministic: false
|
95 |
+
curiosity:
|
96 |
+
gamma: 0.99
|
97 |
+
strength: 0.02
|
98 |
+
network_settings:
|
99 |
+
normalize: false
|
100 |
+
hidden_units: 256
|
101 |
+
num_layers: 2
|
102 |
+
vis_encode_type: simple
|
103 |
+
memory: null
|
104 |
+
goal_conditioning_type: hyper
|
105 |
+
deterministic: false
|
106 |
+
learning_rate: 0.0003
|
107 |
+
encoding_size: null
|
108 |
+
init_path: null
|
109 |
+
keep_checkpoints: 5
|
110 |
+
even_checkpoints: false
|
111 |
+
max_steps: 2000000
|
112 |
+
time_horizon: 512
|
113 |
+
summary_freq: 5000
|
114 |
+
threaded: true
|
115 |
+
self_play: null
|
116 |
+
behavioral_cloning: null
|
117 |
+
env_settings:
|
118 |
+
env_path: ./training-envs-executables/linux/ChopperLnx03/ChopperLnx03
|
119 |
+
env_args: null
|
120 |
+
base_port: 5005
|
121 |
+
num_envs: 1
|
122 |
+
num_areas: 1
|
123 |
+
seed: -1
|
124 |
+
max_lifetime_restarts: 10
|
125 |
+
restarts_rate_limit_n: 1
|
126 |
+
restarts_rate_limit_period_s: 60
|
127 |
+
engine_settings:
|
128 |
+
width: 84
|
129 |
+
height: 84
|
130 |
+
quality_level: 5
|
131 |
+
time_scale: 20
|
132 |
+
target_frame_rate: -1
|
133 |
+
capture_frame_rate: 60
|
134 |
+
no_graphics: true
|
135 |
+
environment_parameters: null
|
136 |
+
checkpoint_settings:
|
137 |
+
run_id: Chopper03_00
|
138 |
+
initialize_from: null
|
139 |
+
load_model: false
|
140 |
+
resume: false
|
141 |
+
force: false
|
142 |
+
train_model: false
|
143 |
+
inference: false
|
144 |
+
results_dir: results
|
145 |
+
torch_settings:
|
146 |
+
device: null
|
147 |
+
debug: false
|
run_logs/Player-0.log
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:62c645e09fea9fb76611de85df0fa65038b5205a10bb0c0131652055ebedfc37
|
3 |
+
size 17480447
|
run_logs/timers.json
ADDED
@@ -0,0 +1,180 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"name": "root",
|
3 |
+
"metadata": {
|
4 |
+
"timer_format_version": "0.1.0",
|
5 |
+
"start_time_seconds": "1691441414",
|
6 |
+
"python_version": "3.10.12 (main, Jun 11 2023, 05:26:28) [GCC 11.4.0]",
|
7 |
+
"command_line_arguments": "/usr/local/bin/mlagents-learn ./config/ppo/Beach.yaml --env=./training-envs-executables/linux/ChopperLnx03/ChopperLnx03 --run-id=Chopper03_00 --no-graphics",
|
8 |
+
"mlagents_version": "0.31.0.dev0",
|
9 |
+
"mlagents_envs_version": "0.31.0.dev0",
|
10 |
+
"communication_protocol_version": "1.5.0",
|
11 |
+
"pytorch_version": "1.11.0+cu102",
|
12 |
+
"numpy_version": "1.21.2",
|
13 |
+
"end_time_seconds": "1691449384"
|
14 |
+
},
|
15 |
+
"total": 7969.14715323,
|
16 |
+
"count": 1,
|
17 |
+
"self": 0.19158281999989413,
|
18 |
+
"children": {
|
19 |
+
"run_training.setup": {
|
20 |
+
"total": 0.05618040300004168,
|
21 |
+
"count": 1,
|
22 |
+
"self": 0.05618040300004168
|
23 |
+
},
|
24 |
+
"TrainerController.start_learning": {
|
25 |
+
"total": 7968.899390007,
|
26 |
+
"count": 1,
|
27 |
+
"self": 20.382382175747807,
|
28 |
+
"children": {
|
29 |
+
"TrainerController._reset_env": {
|
30 |
+
"total": 3.289941060999979,
|
31 |
+
"count": 1,
|
32 |
+
"self": 3.289941060999979
|
33 |
+
},
|
34 |
+
"TrainerController.advance": {
|
35 |
+
"total": 7944.670017396253,
|
36 |
+
"count": 351347,
|
37 |
+
"self": 9.456127876464961,
|
38 |
+
"children": {
|
39 |
+
"env_step": {
|
40 |
+
"total": 7935.213889519788,
|
41 |
+
"count": 351347,
|
42 |
+
"self": 5258.841016150836,
|
43 |
+
"children": {
|
44 |
+
"SubprocessEnvManager._take_step": {
|
45 |
+
"total": 2666.7732618971004,
|
46 |
+
"count": 351347,
|
47 |
+
"self": 55.13522642679436,
|
48 |
+
"children": {
|
49 |
+
"TorchPolicy.evaluate": {
|
50 |
+
"total": 2611.638035470306,
|
51 |
+
"count": 349687,
|
52 |
+
"self": 2611.638035470306
|
53 |
+
}
|
54 |
+
}
|
55 |
+
},
|
56 |
+
"workers": {
|
57 |
+
"total": 9.599611471851006,
|
58 |
+
"count": 351346,
|
59 |
+
"self": 0.0,
|
60 |
+
"children": {
|
61 |
+
"worker_root": {
|
62 |
+
"total": 7928.148748510179,
|
63 |
+
"count": 351346,
|
64 |
+
"is_parallel": true,
|
65 |
+
"self": 4338.461399475137,
|
66 |
+
"children": {
|
67 |
+
"run_training.setup": {
|
68 |
+
"total": 0.0,
|
69 |
+
"count": 0,
|
70 |
+
"is_parallel": true,
|
71 |
+
"self": 0.0,
|
72 |
+
"children": {
|
73 |
+
"steps_from_proto": {
|
74 |
+
"total": 0.006323793999968075,
|
75 |
+
"count": 1,
|
76 |
+
"is_parallel": true,
|
77 |
+
"self": 0.004990665999855537,
|
78 |
+
"children": {
|
79 |
+
"_process_rank_one_or_two_observation": {
|
80 |
+
"total": 0.0013331280001125378,
|
81 |
+
"count": 6,
|
82 |
+
"is_parallel": true,
|
83 |
+
"self": 0.0013331280001125378
|
84 |
+
}
|
85 |
+
}
|
86 |
+
},
|
87 |
+
"UnityEnvironment.step": {
|
88 |
+
"total": 0.04867976399998497,
|
89 |
+
"count": 1,
|
90 |
+
"is_parallel": true,
|
91 |
+
"self": 0.00044687900003737013,
|
92 |
+
"children": {
|
93 |
+
"UnityEnvironment._generate_step_input": {
|
94 |
+
"total": 0.0003752380000037192,
|
95 |
+
"count": 1,
|
96 |
+
"is_parallel": true,
|
97 |
+
"self": 0.0003752380000037192
|
98 |
+
},
|
99 |
+
"communicator.exchange": {
|
100 |
+
"total": 0.04638975899996467,
|
101 |
+
"count": 1,
|
102 |
+
"is_parallel": true,
|
103 |
+
"self": 0.04638975899996467
|
104 |
+
},
|
105 |
+
"steps_from_proto": {
|
106 |
+
"total": 0.0014678879999792116,
|
107 |
+
"count": 1,
|
108 |
+
"is_parallel": true,
|
109 |
+
"self": 0.0004198669999482263,
|
110 |
+
"children": {
|
111 |
+
"_process_rank_one_or_two_observation": {
|
112 |
+
"total": 0.0010480210000309853,
|
113 |
+
"count": 6,
|
114 |
+
"is_parallel": true,
|
115 |
+
"self": 0.0010480210000309853
|
116 |
+
}
|
117 |
+
}
|
118 |
+
}
|
119 |
+
}
|
120 |
+
}
|
121 |
+
}
|
122 |
+
},
|
123 |
+
"UnityEnvironment.step": {
|
124 |
+
"total": 3589.6873490350417,
|
125 |
+
"count": 351345,
|
126 |
+
"is_parallel": true,
|
127 |
+
"self": 152.38421451770046,
|
128 |
+
"children": {
|
129 |
+
"UnityEnvironment._generate_step_input": {
|
130 |
+
"total": 94.76092162134836,
|
131 |
+
"count": 351345,
|
132 |
+
"is_parallel": true,
|
133 |
+
"self": 94.76092162134836
|
134 |
+
},
|
135 |
+
"communicator.exchange": {
|
136 |
+
"total": 2819.983938868857,
|
137 |
+
"count": 351345,
|
138 |
+
"is_parallel": true,
|
139 |
+
"self": 2819.983938868857
|
140 |
+
},
|
141 |
+
"steps_from_proto": {
|
142 |
+
"total": 522.5582740271361,
|
143 |
+
"count": 351345,
|
144 |
+
"is_parallel": true,
|
145 |
+
"self": 149.29340560353978,
|
146 |
+
"children": {
|
147 |
+
"_process_rank_one_or_two_observation": {
|
148 |
+
"total": 373.26486842359634,
|
149 |
+
"count": 2108070,
|
150 |
+
"is_parallel": true,
|
151 |
+
"self": 373.26486842359634
|
152 |
+
}
|
153 |
+
}
|
154 |
+
}
|
155 |
+
}
|
156 |
+
}
|
157 |
+
}
|
158 |
+
}
|
159 |
+
}
|
160 |
+
}
|
161 |
+
}
|
162 |
+
}
|
163 |
+
}
|
164 |
+
},
|
165 |
+
"TrainerController._save_models": {
|
166 |
+
"total": 0.5570493739996891,
|
167 |
+
"count": 1,
|
168 |
+
"self": 0.0034515099996497156,
|
169 |
+
"children": {
|
170 |
+
"RLTrainer._checkpoint": {
|
171 |
+
"total": 0.5535978640000394,
|
172 |
+
"count": 1,
|
173 |
+
"self": 0.5535978640000394
|
174 |
+
}
|
175 |
+
}
|
176 |
+
}
|
177 |
+
}
|
178 |
+
}
|
179 |
+
}
|
180 |
+
}
|
run_logs/training_status.json
ADDED
@@ -0,0 +1,74 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"Heli": {
|
3 |
+
"checkpoints": [
|
4 |
+
{
|
5 |
+
"steps": 249993,
|
6 |
+
"file_path": "results/Chopper03_00/Heli/Heli-249993.onnx",
|
7 |
+
"reward": 11.883715203830175,
|
8 |
+
"creation_time": 1691442859.65759,
|
9 |
+
"auxillary_file_paths": [
|
10 |
+
"results/Chopper03_00/Heli/Heli-249993.pt"
|
11 |
+
]
|
12 |
+
},
|
13 |
+
{
|
14 |
+
"steps": 499984,
|
15 |
+
"file_path": "results/Chopper03_00/Heli/Heli-499984.onnx",
|
16 |
+
"reward": 14.810667872428894,
|
17 |
+
"creation_time": 1691444310.6054258,
|
18 |
+
"auxillary_file_paths": [
|
19 |
+
"results/Chopper03_00/Heli/Heli-499984.pt"
|
20 |
+
]
|
21 |
+
},
|
22 |
+
{
|
23 |
+
"steps": 749884,
|
24 |
+
"file_path": "results/Chopper03_00/Heli/Heli-749884.onnx",
|
25 |
+
"reward": 15.142001390457153,
|
26 |
+
"creation_time": 1691445755.4532797,
|
27 |
+
"auxillary_file_paths": [
|
28 |
+
"results/Chopper03_00/Heli/Heli-749884.pt"
|
29 |
+
]
|
30 |
+
},
|
31 |
+
{
|
32 |
+
"steps": 999718,
|
33 |
+
"file_path": "results/Chopper03_00/Heli/Heli-999718.onnx",
|
34 |
+
"reward": 10.120750904083252,
|
35 |
+
"creation_time": 1691447193.6663072,
|
36 |
+
"auxillary_file_paths": [
|
37 |
+
"results/Chopper03_00/Heli/Heli-999718.pt"
|
38 |
+
]
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"steps": 1249792,
|
42 |
+
"file_path": "results/Chopper03_00/Heli/Heli-1249792.onnx",
|
43 |
+
"reward": null,
|
44 |
+
"creation_time": 1691448604.015588,
|
45 |
+
"auxillary_file_paths": [
|
46 |
+
"results/Chopper03_00/Heli/Heli-1249792.pt"
|
47 |
+
]
|
48 |
+
},
|
49 |
+
{
|
50 |
+
"steps": 1398016,
|
51 |
+
"file_path": "results/Chopper03_00/Heli/Heli-1398016.onnx",
|
52 |
+
"reward": 11.389445361163881,
|
53 |
+
"creation_time": 1691449383.8313968,
|
54 |
+
"auxillary_file_paths": [
|
55 |
+
"results/Chopper03_00/Heli/Heli-1398016.pt"
|
56 |
+
]
|
57 |
+
}
|
58 |
+
],
|
59 |
+
"final_checkpoint": {
|
60 |
+
"steps": 1398016,
|
61 |
+
"file_path": "results/Chopper03_00/Heli.onnx",
|
62 |
+
"reward": 11.389445361163881,
|
63 |
+
"creation_time": 1691449383.8313968,
|
64 |
+
"auxillary_file_paths": [
|
65 |
+
"results/Chopper03_00/Heli/Heli-1398016.pt"
|
66 |
+
]
|
67 |
+
}
|
68 |
+
},
|
69 |
+
"metadata": {
|
70 |
+
"stats_format_version": "0.3.0",
|
71 |
+
"mlagents_version": "0.31.0.dev0",
|
72 |
+
"torch_version": "1.11.0+cu102"
|
73 |
+
}
|
74 |
+
}
|