File size: 12,491 Bytes
1abd4d3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
    "name": "root",
    "metadata": {
        "timer_format_version": "0.1.0",
        "start_time_seconds": "1727308428",
        "python_version": "3.10.12 (main, Sep 11 2024, 15:47:36) [GCC 11.4.0]",
        "command_line_arguments": "/usr/local/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.4.1+cu121",
        "numpy_version": "1.23.5",
        "end_time_seconds": "1727308525"
    },
    "total": 97.17917149100003,
    "count": 1,
    "self": 0.744030861000283,
    "children": {
        "run_training.setup": {
            "total": 0.10714686399978746,
            "count": 1,
            "self": 0.10714686399978746
        },
        "TrainerController.start_learning": {
            "total": 96.32799376599996,
            "count": 1,
            "self": 0.08123873001477477,
            "children": {
                "TrainerController._reset_env": {
                    "total": 3.3975899849997404,
                    "count": 1,
                    "self": 3.3975899849997404
                },
                "TrainerController.advance": {
                    "total": 92.84384844298529,
                    "count": 1836,
                    "self": 0.0971803360012018,
                    "children": {
                        "env_step": {
                            "total": 58.48464374099831,
                            "count": 1836,
                            "self": 52.575160507016335,
                            "children": {
                                "SubprocessEnvManager._take_step": {
                                    "total": 5.861860577988864,
                                    "count": 1836,
                                    "self": 0.2609181839852681,
                                    "children": {
                                        "TorchPolicy.evaluate": {
                                            "total": 5.600942394003596,
                                            "count": 1836,
                                            "self": 5.600942394003596
                                        }
                                    }
                                },
                                "workers": {
                                    "total": 0.04762265599310922,
                                    "count": 1835,
                                    "self": 0.0,
                                    "children": {
                                        "worker_root": {
                                            "total": 96.17611822399294,
                                            "count": 1835,
                                            "is_parallel": true,
                                            "self": 50.12907560199301,
                                            "children": {
                                                "run_training.setup": {
                                                    "total": 0.0,
                                                    "count": 0,
                                                    "is_parallel": true,
                                                    "self": 0.0,
                                                    "children": {
                                                        "steps_from_proto": {
                                                            "total": 0.00539184400031445,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.0017690179997771338,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 0.003622826000537316,
                                                                    "count": 8,
                                                                    "is_parallel": true,
                                                                    "self": 0.003622826000537316
                                                                }
                                                            }
                                                        },
                                                        "UnityEnvironment.step": {
                                                            "total": 0.0714891920001719,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.000912266000341333,
                                                            "children": {
                                                                "UnityEnvironment._generate_step_input": {
                                                                    "total": 0.0006459489995904732,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.0006459489995904732
                                                                },
                                                                "communicator.exchange": {
                                                                    "total": 0.06763687200009372,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.06763687200009372
                                                                },
                                                                "steps_from_proto": {
                                                                    "total": 0.0022941050001463736,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.0005651749997923616,
                                                                    "children": {
                                                                        "_process_rank_one_or_two_observation": {
                                                                            "total": 0.001728930000354012,
                                                                            "count": 8,
                                                                            "is_parallel": true,
                                                                            "self": 0.001728930000354012
                                                                        }
                                                                    }
                                                                }
                                                            }
                                                        }
                                                    }
                                                },
                                                "UnityEnvironment.step": {
                                                    "total": 46.047042621999935,
                                                    "count": 1834,
                                                    "is_parallel": true,
                                                    "self": 1.6778693750052298,
                                                    "children": {
                                                        "UnityEnvironment._generate_step_input": {
                                                            "total": 1.0409775770003762,
                                                            "count": 1834,
                                                            "is_parallel": true,
                                                            "self": 1.0409775770003762
                                                        },
                                                        "communicator.exchange": {
                                                            "total": 39.09812014399995,
                                                            "count": 1834,
                                                            "is_parallel": true,
                                                            "self": 39.09812014399995
                                                        },
                                                        "steps_from_proto": {
                                                            "total": 4.23007552599438,
                                                            "count": 1834,
                                                            "is_parallel": true,
                                                            "self": 1.009514419964944,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 3.220561106029436,
                                                                    "count": 14672,
                                                                    "is_parallel": true,
                                                                    "self": 3.220561106029436
                                                                }
                                                            }
                                                        }
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        },
                        "trainer_advance": {
                            "total": 34.26202436598578,
                            "count": 1835,
                            "self": 0.1077539149837321,
                            "children": {
                                "process_trajectory": {
                                    "total": 5.6261535660019035,
                                    "count": 1835,
                                    "self": 5.6261535660019035
                                },
                                "_update_policy": {
                                    "total": 28.528116885000145,
                                    "count": 7,
                                    "self": 11.201156960001299,
                                    "children": {
                                        "TorchPPOOptimizer.update": {
                                            "total": 17.326959924998846,
                                            "count": 663,
                                            "self": 17.326959924998846
                                        }
                                    }
                                }
                            }
                        }
                    }
                },
                "trainer_threads": {
                    "total": 1.7910001588461455e-06,
                    "count": 1,
                    "self": 1.7910001588461455e-06
                },
                "TrainerController._save_models": {
                    "total": 0.005314816999998584,
                    "count": 1,
                    "self": 0.0003653860003396403,
                    "children": {
                        "RLTrainer._checkpoint": {
                            "total": 0.004949430999658944,
                            "count": 1,
                            "self": 0.004949430999658944
                        }
                    }
                }
            }
        }
    }
}