File size: 18,847 Bytes
e1ca46d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
{
    "name": "root",
    "gauges": {
        "Pyramids.Policy.Entropy.mean": {
            "value": 0.15252816677093506,
            "min": 0.14906516671180725,
            "max": 1.414180874824524,
            "count": 100
        },
        "Pyramids.Policy.Entropy.sum": {
            "value": 4588.04736328125,
            "min": 4464.7998046875,
            "max": 42900.58984375,
            "count": 100
        },
        "Pyramids.Step.mean": {
            "value": 2999890.0,
            "min": 29952.0,
            "max": 2999890.0,
            "count": 100
        },
        "Pyramids.Step.sum": {
            "value": 2999890.0,
            "min": 29952.0,
            "max": 2999890.0,
            "count": 100
        },
        "Pyramids.Policy.ExtrinsicValueEstimate.mean": {
            "value": 0.8182879686355591,
            "min": -0.1800912469625473,
            "max": 0.8639610409736633,
            "count": 100
        },
        "Pyramids.Policy.ExtrinsicValueEstimate.sum": {
            "value": 252.85098266601562,
            "min": -42.68162536621094,
            "max": 264.73980712890625,
            "count": 100
        },
        "Pyramids.Policy.RndValueEstimate.mean": {
            "value": 0.01633444055914879,
            "min": -0.01727745309472084,
            "max": 0.46886947751045227,
            "count": 100
        },
        "Pyramids.Policy.RndValueEstimate.sum": {
            "value": 5.047342300415039,
            "min": -4.9931840896606445,
            "max": 111.12206268310547,
            "count": 100
        },
        "Pyramids.Losses.PolicyLoss.mean": {
            "value": 0.06845543435645393,
            "min": 0.06536668095607585,
            "max": 0.07368367805766238,
            "count": 100
        },
        "Pyramids.Losses.PolicyLoss.sum": {
            "value": 1.026831515346809,
            "min": 0.49879954672351967,
            "max": 1.092246385746608,
            "count": 100
        },
        "Pyramids.Losses.ValueLoss.mean": {
            "value": 0.016203644212024904,
            "min": 0.0006459115332719983,
            "max": 0.0173887597076257,
            "count": 100
        },
        "Pyramids.Losses.ValueLoss.sum": {
            "value": 0.24305466318037358,
            "min": 0.009042761465807975,
            "max": 0.2443290390291105,
            "count": 100
        },
        "Pyramids.Policy.LearningRate.mean": {
            "value": 1.5214728262088886e-06,
            "min": 1.5214728262088886e-06,
            "max": 0.00029838354339596195,
            "count": 100
        },
        "Pyramids.Policy.LearningRate.sum": {
            "value": 2.282209239313333e-05,
            "min": 2.282209239313333e-05,
            "max": 0.0039690933769689005,
            "count": 100
        },
        "Pyramids.Policy.Epsilon.mean": {
            "value": 0.10050712444444444,
            "min": 0.10050712444444444,
            "max": 0.19946118095238097,
            "count": 100
        },
        "Pyramids.Policy.Epsilon.sum": {
            "value": 1.5076068666666667,
            "min": 1.3962282666666668,
            "max": 2.767539433333334,
            "count": 100
        },
        "Pyramids.Policy.Beta.mean": {
            "value": 6.066173200000002e-05,
            "min": 6.066173200000002e-05,
            "max": 0.009946171977142856,
            "count": 100
        },
        "Pyramids.Policy.Beta.sum": {
            "value": 0.0009099259800000002,
            "min": 0.0009099259800000002,
            "max": 0.13231080689000002,
            "count": 100
        },
        "Pyramids.Losses.RNDLoss.mean": {
            "value": 0.00619171280413866,
            "min": 0.006113977171480656,
            "max": 0.5689794421195984,
            "count": 100
        },
        "Pyramids.Losses.RNDLoss.sum": {
            "value": 0.09287568926811218,
            "min": 0.08559568226337433,
            "max": 3.982856035232544,
            "count": 100
        },
        "Pyramids.Environment.EpisodeLength.mean": {
            "value": 211.73333333333332,
            "min": 210.13138686131387,
            "max": 999.0,
            "count": 100
        },
        "Pyramids.Environment.EpisodeLength.sum": {
            "value": 31760.0,
            "min": 15984.0,
            "max": 33377.0,
            "count": 100
        },
        "Pyramids.Environment.CumulativeReward.mean": {
            "value": 1.73392078820491,
            "min": -1.0000000521540642,
            "max": 1.7871792959755866,
            "count": 100
        },
        "Pyramids.Environment.CumulativeReward.sum": {
            "value": 258.3541974425316,
            "min": -30.354801654815674,
            "max": 259.14099791646004,
            "count": 100
        },
        "Pyramids.Policy.ExtrinsicReward.mean": {
            "value": 1.73392078820491,
            "min": -1.0000000521540642,
            "max": 1.7871792959755866,
            "count": 100
        },
        "Pyramids.Policy.ExtrinsicReward.sum": {
            "value": 258.3541974425316,
            "min": -30.354801654815674,
            "max": 259.14099791646004,
            "count": 100
        },
        "Pyramids.Policy.RndReward.mean": {
            "value": 0.013677320852500681,
            "min": 0.013677320852500681,
            "max": 11.693298460915685,
            "count": 100
        },
        "Pyramids.Policy.RndReward.sum": {
            "value": 2.0379208070226014,
            "min": 1.8333403850265313,
            "max": 187.09277537465096,
            "count": 100
        },
        "Pyramids.IsTraining.mean": {
            "value": 1.0,
            "min": 1.0,
            "max": 1.0,
            "count": 100
        },
        "Pyramids.IsTraining.sum": {
            "value": 1.0,
            "min": 1.0,
            "max": 1.0,
            "count": 100
        }
    },
    "metadata": {
        "timer_format_version": "0.1.0",
        "start_time_seconds": "1705560733",
        "python_version": "3.10.12 (main, Jan 18 2024, 15:14:31) [GCC 11.4.0]",
        "command_line_arguments": "/home/omar/.pyenv/versions/3.10.12/envs/rlhf/bin/mlagents-learn ./config/ppo/PyramidsRND.yaml --env=./training-envs-executables/linux/Pyramids/Pyramids --run-id=Pyramids Training --no-graphics",
        "mlagents_version": "1.1.0.dev0",
        "mlagents_envs_version": "1.1.0.dev0",
        "communication_protocol_version": "1.5.0",
        "pytorch_version": "2.1.2+cu121",
        "numpy_version": "1.23.5",
        "end_time_seconds": "1705562784"
    },
    "total": 2051.1559370680006,
    "count": 1,
    "self": 0.16801876400131732,
    "children": {
        "run_training.setup": {
            "total": 0.010624038000059954,
            "count": 1,
            "self": 0.010624038000059954
        },
        "TrainerController.start_learning": {
            "total": 2050.9772942659993,
            "count": 1,
            "self": 1.6147928751834115,
            "children": {
                "TrainerController._reset_env": {
                    "total": 0.6259975359998862,
                    "count": 1,
                    "self": 0.6259975359998862
                },
                "TrainerController.advance": {
                    "total": 2048.6991612388165,
                    "count": 195036,
                    "self": 1.4775207274569766,
                    "children": {
                        "env_step": {
                            "total": 1287.8277061731478,
                            "count": 195036,
                            "self": 1166.8335389201993,
                            "children": {
                                "SubprocessEnvManager._take_step": {
                                    "total": 119.98410006001086,
                                    "count": 195036,
                                    "self": 5.1350317988071765,
                                    "children": {
                                        "TorchPolicy.evaluate": {
                                            "total": 114.84906826120368,
                                            "count": 187567,
                                            "self": 114.84906826120368
                                        }
                                    }
                                },
                                "workers": {
                                    "total": 1.0100671929376404,
                                    "count": 195036,
                                    "self": 0.0,
                                    "children": {
                                        "worker_root": {
                                            "total": 2048.7941494200104,
                                            "count": 195036,
                                            "is_parallel": true,
                                            "self": 995.055227156945,
                                            "children": {
                                                "run_training.setup": {
                                                    "total": 0.0,
                                                    "count": 0,
                                                    "is_parallel": true,
                                                    "self": 0.0,
                                                    "children": {
                                                        "steps_from_proto": {
                                                            "total": 0.000967019000199798,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.00026698600140662165,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 0.0007000329987931764,
                                                                    "count": 8,
                                                                    "is_parallel": true,
                                                                    "self": 0.0007000329987931764
                                                                }
                                                            }
                                                        },
                                                        "UnityEnvironment.step": {
                                                            "total": 0.01505296899995301,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.0001391280002280837,
                                                            "children": {
                                                                "UnityEnvironment._generate_step_input": {
                                                                    "total": 0.00020458800008782418,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.00020458800008782418
                                                                },
                                                                "communicator.exchange": {
                                                                    "total": 0.01417733899961604,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.01417733899961604
                                                                },
                                                                "steps_from_proto": {
                                                                    "total": 0.0005319140000210609,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.00015708800128777511,
                                                                    "children": {
                                                                        "_process_rank_one_or_two_observation": {
                                                                            "total": 0.0003748259987332858,
                                                                            "count": 8,
                                                                            "is_parallel": true,
                                                                            "self": 0.0003748259987332858
                                                                        }
                                                                    }
                                                                }
                                                            }
                                                        }
                                                    }
                                                },
                                                "UnityEnvironment.step": {
                                                    "total": 1053.7389222630654,
                                                    "count": 195035,
                                                    "is_parallel": true,
                                                    "self": 20.99056299007316,
                                                    "children": {
                                                        "UnityEnvironment._generate_step_input": {
                                                            "total": 17.357348764071503,
                                                            "count": 195035,
                                                            "is_parallel": true,
                                                            "self": 17.357348764071503
                                                        },
                                                        "communicator.exchange": {
                                                            "total": 947.2814714219066,
                                                            "count": 195035,
                                                            "is_parallel": true,
                                                            "self": 947.2814714219066
                                                        },
                                                        "steps_from_proto": {
                                                            "total": 68.10953908701413,
                                                            "count": 195035,
                                                            "is_parallel": true,
                                                            "self": 14.092228908545621,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 54.017310178468506,
                                                                    "count": 1560280,
                                                                    "is_parallel": true,
                                                                    "self": 54.017310178468506
                                                                }
                                                            }
                                                        }
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        },
                        "trainer_advance": {
                            "total": 759.3939343382117,
                            "count": 195036,
                            "self": 3.448474896033076,
                            "children": {
                                "process_trajectory": {
                                    "total": 163.58176201317292,
                                    "count": 195036,
                                    "self": 163.32704501017452,
                                    "children": {
                                        "RLTrainer._checkpoint": {
                                            "total": 0.2547170029984045,
                                            "count": 6,
                                            "self": 0.2547170029984045
                                        }
                                    }
                                },
                                "_update_policy": {
                                    "total": 592.3636974290057,
                                    "count": 1398,
                                    "self": 391.9358900801535,
                                    "children": {
                                        "TorchPPOOptimizer.update": {
                                            "total": 200.42780734885218,
                                            "count": 68391,
                                            "self": 200.42780734885218
                                        }
                                    }
                                }
                            }
                        }
                    }
                },
                "trainer_threads": {
                    "total": 5.800002327305265e-07,
                    "count": 1,
                    "self": 5.800002327305265e-07
                },
                "TrainerController._save_models": {
                    "total": 0.03734203599924513,
                    "count": 1,
                    "self": 0.0009203109993904945,
                    "children": {
                        "RLTrainer._checkpoint": {
                            "total": 0.03642172499985463,
                            "count": 1,
                            "self": 0.03642172499985463
                        }
                    }
                }
            }
        }
    }
}