File size: 12,557 Bytes
0b08716
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "1789141089",
        "python_version": "3.10.12 (main, Jul  5 2023, 18:54:27) [GCC 11.2.0]",
        "command_line_arguments": "/content/miniconda/envs/mlagents/bin/mlagents-learn /content/ml-agents/config/ppo/PyramidsRND.yaml --env=/content/ml-agents/training-envs-executables/linux/Pyramids/Pyramids --run-id=Pyramids5K --no-graphics --torch-device=cuda",
        "mlagents_version": "1.2.0.dev0",
        "mlagents_envs_version": "1.2.0.dev0",
        "communication_protocol_version": "1.5.0",
        "pytorch_version": "2.8.0+cu128",
        "numpy_version": "1.23.5",
        "end_time_seconds": "1789141103"
    },
    "total": 14.320735654999226,
    "count": 1,
    "self": 0.4357702790002804,
    "children": {
        "run_training.setup": {
            "total": 0.019506590999299078,
            "count": 1,
            "self": 0.019506590999299078
        },
        "TrainerController.start_learning": {
            "total": 13.865458784999646,
            "count": 1,
            "self": 0.007325101001697476,
            "children": {
                "TrainerController._reset_env": {
                    "total": 2.142818628999521,
                    "count": 1,
                    "self": 2.142818628999521
                },
                "TrainerController.advance": {
                    "total": 11.608379001996582,
                    "count": 384,
                    "self": 0.007890972014138242,
                    "children": {
                        "env_step": {
                            "total": 8.607647524997446,
                            "count": 384,
                            "self": 7.543522124988158,
                            "children": {
                                "SubprocessEnvManager._take_step": {
                                    "total": 1.0596707680051622,
                                    "count": 384,
                                    "self": 0.028197604968227097,
                                    "children": {
                                        "TorchPolicy.evaluate": {
                                            "total": 1.0314731630369351,
                                            "count": 384,
                                            "self": 1.0314731630369351
                                        }
                                    }
                                },
                                "workers": {
                                    "total": 0.004454632004126324,
                                    "count": 384,
                                    "self": 0.0,
                                    "children": {
                                        "worker_root": {
                                            "total": 13.576688503981131,
                                            "count": 384,
                                            "is_parallel": true,
                                            "self": 6.894392600977881,
                                            "children": {
                                                "run_training.setup": {
                                                    "total": 0.0,
                                                    "count": 0,
                                                    "is_parallel": true,
                                                    "self": 0.0,
                                                    "children": {
                                                        "steps_from_proto": {
                                                            "total": 0.0018168149999837624,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.0006790940005885204,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 0.001137720999395242,
                                                                    "count": 8,
                                                                    "is_parallel": true,
                                                                    "self": 0.001137720999395242
                                                                }
                                                            }
                                                        },
                                                        "UnityEnvironment.step": {
                                                            "total": 0.07183210699986375,
                                                            "count": 1,
                                                            "is_parallel": true,
                                                            "self": 0.0006148449974716641,
                                                            "children": {
                                                                "UnityEnvironment._generate_step_input": {
                                                                    "total": 0.0004242530012561474,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.0004242530012561474
                                                                },
                                                                "communicator.exchange": {
                                                                    "total": 0.0691189220015076,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.0691189220015076
                                                                },
                                                                "steps_from_proto": {
                                                                    "total": 0.0016740869996283436,
                                                                    "count": 1,
                                                                    "is_parallel": true,
                                                                    "self": 0.0003674259987747064,
                                                                    "children": {
                                                                        "_process_rank_one_or_two_observation": {
                                                                            "total": 0.0013066610008536372,
                                                                            "count": 8,
                                                                            "is_parallel": true,
                                                                            "self": 0.0013066610008536372
                                                                        }
                                                                    }
                                                                }
                                                            }
                                                        }
                                                    }
                                                },
                                                "UnityEnvironment.step": {
                                                    "total": 6.682295903003251,
                                                    "count": 383,
                                                    "is_parallel": true,
                                                    "self": 0.20823834696420818,
                                                    "children": {
                                                        "UnityEnvironment._generate_step_input": {
                                                            "total": 0.1324775670273084,
                                                            "count": 383,
                                                            "is_parallel": true,
                                                            "self": 0.1324775670273084
                                                        },
                                                        "communicator.exchange": {
                                                            "total": 5.63814970801468,
                                                            "count": 383,
                                                            "is_parallel": true,
                                                            "self": 5.63814970801468
                                                        },
                                                        "steps_from_proto": {
                                                            "total": 0.7034302809970541,
                                                            "count": 383,
                                                            "is_parallel": true,
                                                            "self": 0.14115657600450504,
                                                            "children": {
                                                                "_process_rank_one_or_two_observation": {
                                                                    "total": 0.562273704992549,
                                                                    "count": 3064,
                                                                    "is_parallel": true,
                                                                    "self": 0.562273704992549
                                                                }
                                                            }
                                                        }
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        },
                        "trainer_advance": {
                            "total": 2.992840504984997,
                            "count": 384,
                            "self": 0.009308433976912056,
                            "children": {
                                "process_trajectory": {
                                    "total": 0.6437136150088918,
                                    "count": 384,
                                    "self": 0.6437136150088918
                                },
                                "_update_policy": {
                                    "total": 2.3398184559991932,
                                    "count": 1,
                                    "self": 1.09186908900665,
                                    "children": {
                                        "TorchPPOOptimizer.update": {
                                            "total": 1.2479493669925432,
                                            "count": 96,
                                            "self": 1.2479493669925432
                                        }
                                    }
                                }
                            }
                        }
                    }
                },
                "trainer_threads": {
                    "total": 9.760005923453718e-07,
                    "count": 1,
                    "self": 9.760005923453718e-07
                },
                "TrainerController._save_models": {
                    "total": 0.10693507700125338,
                    "count": 1,
                    "self": 0.001001498001642176,
                    "children": {
                        "RLTrainer._checkpoint": {
                            "total": 0.1059335789996112,
                            "count": 1,
                            "self": 0.1059335789996112
                        }
                    }
                }
            }
        }
    }
}