PaulTbbr commited on
Commit
fe1318d
·
verified ·
1 Parent(s): 0732cab

First push

Browse files
Pyramids.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9ea443e4f8e6d278d63376fcf4beb0b1cb86452abc22c3296356fb411babdad
3
+ size 1420238
Pyramids/Pyramids-10000044.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9ea443e4f8e6d278d63376fcf4beb0b1cb86452abc22c3296356fb411babdad
3
+ size 1420238
Pyramids/Pyramids-10000044.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2f527d951b9034c61cfa6bb15a8cbe9e49ec20299e40eb24d5f6ab26e032ff0
3
+ size 10002123
Pyramids/Pyramids-8499966.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:346e4933461777237cc962fb92bc6f9db59f58db052fdf6059e68c50d19ddcb7
3
+ size 1420238
Pyramids/Pyramids-8499966.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71bf416543947247028829e056fae96f4a782a133561a1546eff65b21ed09e9a
3
+ size 10002044
Pyramids/Pyramids-8999901.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fbef9d0164eeba1bdc26d115848d938dbd1a2e81c50d57916d4401612d1e862
3
+ size 1420238
Pyramids/Pyramids-8999901.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea4376c9ceed196000866144b3316bc6138946eb431490b694fc4657d3e42be6
3
+ size 10002044
Pyramids/Pyramids-9499969.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1048e0768cb55a793aa76d5db55cb1687970c73013b3b25a80b6fb2c530f8e1e
3
+ size 1420238
Pyramids/Pyramids-9499969.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f350dfe74520b632bd4bff6fcb582a2e7ec759349d15c0c91dfa603a3e0d121
3
+ size 10002044
Pyramids/Pyramids-9999916.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9ea443e4f8e6d278d63376fcf4beb0b1cb86452abc22c3296356fb411babdad
3
+ size 1420238
Pyramids/Pyramids-9999916.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50a7222fecf2fd8c43419d6d8e6e3183305cb87688b66b51e2e9d85b80ca2b08
3
+ size 10002044
Pyramids/checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42e37f046bdd75fce87240b1f503bc1845eb49d9b43ee533077396fd2a7b3e53
3
+ size 10001506
Pyramids/events.out.tfevents.1711138902.MacBook-Pro-de-Paul.local.15473.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a218097a52a5d93905ed71c438eaf18e4789e1860f363b3b9c8eb1fe2836ba6
3
+ size 2723399
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: ml-agents
3
+ tags:
4
+ - Pyramids
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - ML-Agents-Pyramids
8
+ ---
9
+
10
+ # **ppo** Agent playing **Pyramids**
11
+ This is a trained model of a **ppo** agent playing **Pyramids**
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: PaulTbbr/ppo-pyramids-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": {"Pyramids": {"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": 10000000, "time_horizon": 128, "summary_freq": 30000, "threaded": false, "self_play": null, "behavioral_cloning": null}}, "env_settings": {"env_path": null, "env_args": null, "base_port": 5005, "num_envs": 1, "num_areas": 1, "timeout_wait": 60, "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": false, "no_graphics_monitor": false}, "environment_parameters": null, "checkpoint_settings": {"run_id": "Pyramids00", "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,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_settings: null
2
+ behaviors:
3
+ Pyramids:
4
+ trainer_type: ppo
5
+ hyperparameters:
6
+ batch_size: 128
7
+ buffer_size: 2048
8
+ learning_rate: 0.0003
9
+ beta: 0.01
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: 500000
18
+ network_settings:
19
+ normalize: false
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: 5
53
+ even_checkpoints: false
54
+ max_steps: 10000000
55
+ time_horizon: 128
56
+ summary_freq: 30000
57
+ threaded: false
58
+ self_play: null
59
+ behavioral_cloning: null
60
+ env_settings:
61
+ env_path: null
62
+ env_args: null
63
+ base_port: 5005
64
+ num_envs: 1
65
+ num_areas: 1
66
+ timeout_wait: 60
67
+ seed: -1
68
+ max_lifetime_restarts: 10
69
+ restarts_rate_limit_n: 1
70
+ restarts_rate_limit_period_s: 60
71
+ engine_settings:
72
+ width: 84
73
+ height: 84
74
+ quality_level: 5
75
+ time_scale: 20
76
+ target_frame_rate: -1
77
+ capture_frame_rate: 60
78
+ no_graphics: false
79
+ no_graphics_monitor: false
80
+ environment_parameters: null
81
+ checkpoint_settings:
82
+ run_id: Pyramids00
83
+ initialize_from: null
84
+ load_model: false
85
+ resume: false
86
+ force: false
87
+ train_model: false
88
+ inference: false
89
+ results_dir: results
90
+ torch_settings:
91
+ device: null
92
+ debug: false
run_logs/timers.json ADDED
@@ -0,0 +1,368 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "root",
3
+ "gauges": {
4
+ "Pyramids.Policy.Entropy.mean": {
5
+ "value": 0.09659828245639801,
6
+ "min": 0.08577015995979309,
7
+ "max": 1.4727826118469238,
8
+ "count": 333
9
+ },
10
+ "Pyramids.Policy.Entropy.sum": {
11
+ "value": 2897.948486328125,
12
+ "min": 2615.646728515625,
13
+ "max": 44678.33203125,
14
+ "count": 333
15
+ },
16
+ "Pyramids.Step.mean": {
17
+ "value": 9989950.0,
18
+ "min": 29952.0,
19
+ "max": 9989950.0,
20
+ "count": 333
21
+ },
22
+ "Pyramids.Step.sum": {
23
+ "value": 9989950.0,
24
+ "min": 29952.0,
25
+ "max": 9989950.0,
26
+ "count": 333
27
+ },
28
+ "Pyramids.Policy.ExtrinsicValueEstimate.mean": {
29
+ "value": 0.9547367691993713,
30
+ "min": -0.10988577455282211,
31
+ "max": 1.004745602607727,
32
+ "count": 333
33
+ },
34
+ "Pyramids.Policy.ExtrinsicValueEstimate.sum": {
35
+ "value": 299.787353515625,
36
+ "min": -26.482471466064453,
37
+ "max": 323.528076171875,
38
+ "count": 333
39
+ },
40
+ "Pyramids.Policy.CuriosityValueEstimate.mean": {
41
+ "value": 0.030147476121783257,
42
+ "min": 0.025316940620541573,
43
+ "max": 0.41487738490104675,
44
+ "count": 333
45
+ },
46
+ "Pyramids.Policy.CuriosityValueEstimate.sum": {
47
+ "value": 9.466307640075684,
48
+ "min": 7.822934627532959,
49
+ "max": 99.82791137695312,
50
+ "count": 333
51
+ },
52
+ "Pyramids.Losses.PolicyLoss.mean": {
53
+ "value": 0.0673175727230652,
54
+ "min": 0.06232600730644273,
55
+ "max": 0.07484725077032427,
56
+ "count": 333
57
+ },
58
+ "Pyramids.Losses.PolicyLoss.sum": {
59
+ "value": 0.9424460181229127,
60
+ "min": 0.47229922364479937,
61
+ "max": 1.1045833782774541,
62
+ "count": 333
63
+ },
64
+ "Pyramids.Losses.ValueLoss.mean": {
65
+ "value": 0.013726562569943436,
66
+ "min": 6.54922021441563e-05,
67
+ "max": 0.017306548792735806,
68
+ "count": 333
69
+ },
70
+ "Pyramids.Losses.ValueLoss.sum": {
71
+ "value": 0.1921718759792081,
72
+ "min": 0.0009168908300181881,
73
+ "max": 0.24229168309830126,
74
+ "count": 333
75
+ },
76
+ "Pyramids.Policy.LearningRate.mean": {
77
+ "value": 7.38710468082143e-07,
78
+ "min": 7.38710468082143e-07,
79
+ "max": 0.0002995150630187886,
80
+ "count": 333
81
+ },
82
+ "Pyramids.Policy.LearningRate.sum": {
83
+ "value": 1.0341946553150002e-05,
84
+ "min": 1.0341946553150002e-05,
85
+ "max": 0.0040343232352256396,
86
+ "count": 333
87
+ },
88
+ "Pyramids.Policy.Epsilon.mean": {
89
+ "value": 0.10024620357142858,
90
+ "min": 0.10024620357142858,
91
+ "max": 0.1998383542857143,
92
+ "count": 333
93
+ },
94
+ "Pyramids.Policy.Epsilon.sum": {
95
+ "value": 1.4034468500000001,
96
+ "min": 1.38850944,
97
+ "max": 2.8447743599999997,
98
+ "count": 333
99
+ },
100
+ "Pyramids.Policy.Beta.mean": {
101
+ "value": 3.45957367857143e-05,
102
+ "min": 3.45957367857143e-05,
103
+ "max": 0.009983851593142858,
104
+ "count": 333
105
+ },
106
+ "Pyramids.Policy.Beta.sum": {
107
+ "value": 0.0004843403150000002,
108
+ "min": 0.0004843403150000002,
109
+ "max": 0.134492958564,
110
+ "count": 333
111
+ },
112
+ "Pyramids.Losses.CuriosityForwardLoss.mean": {
113
+ "value": 0.010238115377779605,
114
+ "min": 0.008714325094409841,
115
+ "max": 0.4099465698968735,
116
+ "count": 333
117
+ },
118
+ "Pyramids.Losses.CuriosityForwardLoss.sum": {
119
+ "value": 0.14333361528891447,
120
+ "min": 0.12200055132173777,
121
+ "max": 2.8696259892781146,
122
+ "count": 333
123
+ },
124
+ "Pyramids.Losses.CuriosityInverseLoss.mean": {
125
+ "value": 0.0057052407149577825,
126
+ "min": 0.002950076695603659,
127
+ "max": 0.7842482853396406,
128
+ "count": 333
129
+ },
130
+ "Pyramids.Losses.CuriosityInverseLoss.sum": {
131
+ "value": 0.07987337000940896,
132
+ "min": 0.044251150434054885,
133
+ "max": 5.4897379973774845,
134
+ "count": 333
135
+ },
136
+ "Pyramids.Environment.EpisodeLength.mean": {
137
+ "value": 186.1556886227545,
138
+ "min": 171.22085889570553,
139
+ "max": 999.0,
140
+ "count": 333
141
+ },
142
+ "Pyramids.Environment.EpisodeLength.sum": {
143
+ "value": 31088.0,
144
+ "min": 15984.0,
145
+ "max": 34211.0,
146
+ "count": 333
147
+ },
148
+ "Pyramids.Environment.CumulativeReward.mean": {
149
+ "value": 1.7891301097550305,
150
+ "min": -1.0000000521540642,
151
+ "max": 1.8287791264203428,
152
+ "count": 333
153
+ },
154
+ "Pyramids.Environment.CumulativeReward.sum": {
155
+ "value": 296.9955982193351,
156
+ "min": -32.000001668930054,
157
+ "max": 314.61059732735157,
158
+ "count": 333
159
+ },
160
+ "Pyramids.Policy.ExtrinsicReward.mean": {
161
+ "value": 1.7891301097550305,
162
+ "min": -1.0000000521540642,
163
+ "max": 1.8287791264203428,
164
+ "count": 333
165
+ },
166
+ "Pyramids.Policy.ExtrinsicReward.sum": {
167
+ "value": 296.9955982193351,
168
+ "min": -32.000001668930054,
169
+ "max": 314.61059732735157,
170
+ "count": 333
171
+ },
172
+ "Pyramids.Policy.CuriosityReward.mean": {
173
+ "value": 0.03991356181367917,
174
+ "min": 0.03410876181021051,
175
+ "max": 7.369642803445458,
176
+ "count": 333
177
+ },
178
+ "Pyramids.Policy.CuriosityReward.sum": {
179
+ "value": 6.625651261070743,
180
+ "min": 5.184531795151997,
181
+ "max": 167.39837768673897,
182
+ "count": 333
183
+ },
184
+ "Pyramids.IsTraining.mean": {
185
+ "value": 1.0,
186
+ "min": 1.0,
187
+ "max": 1.0,
188
+ "count": 333
189
+ },
190
+ "Pyramids.IsTraining.sum": {
191
+ "value": 1.0,
192
+ "min": 1.0,
193
+ "max": 1.0,
194
+ "count": 333
195
+ }
196
+ },
197
+ "metadata": {
198
+ "timer_format_version": "0.1.0",
199
+ "start_time_seconds": "1711138897",
200
+ "python_version": "3.10.1 (main, Mar 20 2024, 23:11:47) [Clang 15.0.0 (clang-1500.3.9.4)]",
201
+ "command_line_arguments": "/Users/paul/.pyenv/versions/3.10.1/envs/venv_unit5_hf/bin/mlagents-learn config/ppo/Pyramids.yaml --run-id Pyramids00",
202
+ "mlagents_version": "1.1.0.dev0",
203
+ "mlagents_envs_version": "1.1.0.dev0",
204
+ "communication_protocol_version": "1.5.0",
205
+ "pytorch_version": "2.2.1",
206
+ "numpy_version": "1.23.5",
207
+ "end_time_seconds": "1711155413"
208
+ },
209
+ "total": 16515.837095,
210
+ "count": 1,
211
+ "self": 0.0032615419950161595,
212
+ "children": {
213
+ "run_training.setup": {
214
+ "total": 0.014169707999826642,
215
+ "count": 1,
216
+ "self": 0.014169707999826642
217
+ },
218
+ "TrainerController.start_learning": {
219
+ "total": 16515.819663750004,
220
+ "count": 1,
221
+ "self": 5.050972458069737,
222
+ "children": {
223
+ "TrainerController._reset_env": {
224
+ "total": 5.200303416000679,
225
+ "count": 1,
226
+ "self": 5.200303416000679
227
+ },
228
+ "TrainerController.advance": {
229
+ "total": 16505.53490170994,
230
+ "count": 655734,
231
+ "self": 5.053516233303526,
232
+ "children": {
233
+ "env_step": {
234
+ "total": 13992.83843329976,
235
+ "count": 655734,
236
+ "self": 13668.814883681367,
237
+ "children": {
238
+ "SubprocessEnvManager._take_step": {
239
+ "total": 320.4675482351122,
240
+ "count": 655734,
241
+ "self": 11.787854892958421,
242
+ "children": {
243
+ "TorchPolicy.evaluate": {
244
+ "total": 308.67969334215377,
245
+ "count": 625052,
246
+ "self": 308.67969334215377
247
+ }
248
+ }
249
+ },
250
+ "workers": {
251
+ "total": 3.5560013832800905,
252
+ "count": 655734,
253
+ "self": 0.0,
254
+ "children": {
255
+ "worker_root": {
256
+ "total": 16503.084301704188,
257
+ "count": 655734,
258
+ "is_parallel": true,
259
+ "self": 3187.134702260697,
260
+ "children": {
261
+ "steps_from_proto": {
262
+ "total": 0.0005815830008941703,
263
+ "count": 1,
264
+ "is_parallel": true,
265
+ "self": 0.00012183199942228384,
266
+ "children": {
267
+ "_process_rank_one_or_two_observation": {
268
+ "total": 0.0004597510014718864,
269
+ "count": 8,
270
+ "is_parallel": true,
271
+ "self": 0.0004597510014718864
272
+ }
273
+ }
274
+ },
275
+ "UnityEnvironment.step": {
276
+ "total": 13315.94901786049,
277
+ "count": 655734,
278
+ "is_parallel": true,
279
+ "self": 39.6326328279647,
280
+ "children": {
281
+ "UnityEnvironment._generate_step_input": {
282
+ "total": 185.1218753106259,
283
+ "count": 655734,
284
+ "is_parallel": true,
285
+ "self": 185.1218753106259
286
+ },
287
+ "communicator.exchange": {
288
+ "total": 12774.521226542933,
289
+ "count": 655734,
290
+ "is_parallel": true,
291
+ "self": 12774.521226542933
292
+ },
293
+ "steps_from_proto": {
294
+ "total": 316.6732831789668,
295
+ "count": 655734,
296
+ "is_parallel": true,
297
+ "self": 55.268841222059564,
298
+ "children": {
299
+ "_process_rank_one_or_two_observation": {
300
+ "total": 261.40444195690725,
301
+ "count": 5245872,
302
+ "is_parallel": true,
303
+ "self": 261.40444195690725
304
+ }
305
+ }
306
+ }
307
+ }
308
+ }
309
+ }
310
+ }
311
+ }
312
+ }
313
+ }
314
+ },
315
+ "trainer_advance": {
316
+ "total": 2507.642952176877,
317
+ "count": 655734,
318
+ "self": 9.581380121315306,
319
+ "children": {
320
+ "process_trajectory": {
321
+ "total": 436.2491943975365,
322
+ "count": 655734,
323
+ "self": 435.53307568854143,
324
+ "children": {
325
+ "RLTrainer._checkpoint": {
326
+ "total": 0.7161187089950545,
327
+ "count": 20,
328
+ "self": 0.7161187089950545
329
+ }
330
+ }
331
+ },
332
+ "_update_policy": {
333
+ "total": 2061.812377658025,
334
+ "count": 4649,
335
+ "self": 1470.146524662432,
336
+ "children": {
337
+ "TorchPPOOptimizer.update": {
338
+ "total": 591.665852995593,
339
+ "count": 227991,
340
+ "self": 591.665852995593
341
+ }
342
+ }
343
+ }
344
+ }
345
+ }
346
+ }
347
+ },
348
+ "trainer_threads": {
349
+ "total": 3.749955794773996e-07,
350
+ "count": 1,
351
+ "self": 3.749955794773996e-07
352
+ },
353
+ "TrainerController._save_models": {
354
+ "total": 0.03348579099838389,
355
+ "count": 1,
356
+ "self": 0.0004614159988705069,
357
+ "children": {
358
+ "RLTrainer._checkpoint": {
359
+ "total": 0.033024374999513384,
360
+ "count": 1,
361
+ "self": 0.033024374999513384
362
+ }
363
+ }
364
+ }
365
+ }
366
+ }
367
+ }
368
+ }
run_logs/training_status.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "Pyramids": {
3
+ "checkpoints": [
4
+ {
5
+ "steps": 8499966,
6
+ "file_path": "results/Pyramids00/Pyramids/Pyramids-8499966.onnx",
7
+ "reward": null,
8
+ "creation_time": 1711152905.320728,
9
+ "auxillary_file_paths": [
10
+ "results/Pyramids00/Pyramids/Pyramids-8499966.pt"
11
+ ]
12
+ },
13
+ {
14
+ "steps": 8999901,
15
+ "file_path": "results/Pyramids00/Pyramids/Pyramids-8999901.onnx",
16
+ "reward": 1.8924999833106995,
17
+ "creation_time": 1711153744.818826,
18
+ "auxillary_file_paths": [
19
+ "results/Pyramids00/Pyramids/Pyramids-8999901.pt"
20
+ ]
21
+ },
22
+ {
23
+ "steps": 9499969,
24
+ "file_path": "results/Pyramids00/Pyramids/Pyramids-9499969.onnx",
25
+ "reward": 1.5809999711811542,
26
+ "creation_time": 1711154574.8163252,
27
+ "auxillary_file_paths": [
28
+ "results/Pyramids00/Pyramids/Pyramids-9499969.pt"
29
+ ]
30
+ },
31
+ {
32
+ "steps": 9999916,
33
+ "file_path": "results/Pyramids00/Pyramids/Pyramids-9999916.onnx",
34
+ "reward": 1.8183333228031795,
35
+ "creation_time": 1711155413.550189,
36
+ "auxillary_file_paths": [
37
+ "results/Pyramids00/Pyramids/Pyramids-9999916.pt"
38
+ ]
39
+ },
40
+ {
41
+ "steps": 10000044,
42
+ "file_path": "results/Pyramids00/Pyramids/Pyramids-10000044.onnx",
43
+ "reward": 1.8183333228031795,
44
+ "creation_time": 1711155413.587848,
45
+ "auxillary_file_paths": [
46
+ "results/Pyramids00/Pyramids/Pyramids-10000044.pt"
47
+ ]
48
+ }
49
+ ],
50
+ "final_checkpoint": {
51
+ "steps": 10000044,
52
+ "file_path": "results/Pyramids00/Pyramids.onnx",
53
+ "reward": 1.8183333228031795,
54
+ "creation_time": 1711155413.587848,
55
+ "auxillary_file_paths": [
56
+ "results/Pyramids00/Pyramids/Pyramids-10000044.pt"
57
+ ]
58
+ }
59
+ },
60
+ "metadata": {
61
+ "stats_format_version": "0.3.0",
62
+ "mlagents_version": "1.1.0.dev0",
63
+ "torch_version": "2.2.1"
64
+ }
65
+ }