yepengsun's picture
First Push
5bc474e verified
{
"name": "root",
"metadata": {
"timer_format_version": "0.1.0",
"start_time_seconds": "1757641723",
"python_version": "3.10.12 (main, Jul 5 2023, 18:54:27) [GCC 11.2.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.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": "1757641760"
},
"total": 36.4317104859997,
"count": 1,
"self": 0.21089216100017438,
"children": {
"run_training.setup": {
"total": 0.022395624999717256,
"count": 1,
"self": 0.022395624999717256
},
"TrainerController.start_learning": {
"total": 36.19842269999981,
"count": 1,
"self": 0.02610622400379725,
"children": {
"TrainerController._reset_env": {
"total": 2.103652128000249,
"count": 1,
"self": 2.103652128000249
},
"TrainerController.advance": {
"total": 33.95796223799607,
"count": 1492,
"self": 0.026589855007841834,
"children": {
"env_step": {
"total": 21.06942381398676,
"count": 1492,
"self": 17.71239265198801,
"children": {
"SubprocessEnvManager._take_step": {
"total": 3.3411463689949414,
"count": 1492,
"self": 0.0992639009823506,
"children": {
"TorchPolicy.evaluate": {
"total": 3.241882468012591,
"count": 1492,
"self": 3.241882468012591
}
}
},
"workers": {
"total": 0.015884793003806408,
"count": 1491,
"self": 0.0,
"children": {
"worker_root": {
"total": 36.04512019600725,
"count": 1491,
"is_parallel": true,
"self": 20.311299017993406,
"children": {
"run_training.setup": {
"total": 0.0,
"count": 0,
"is_parallel": true,
"self": 0.0,
"children": {
"steps_from_proto": {
"total": 0.0018522639998082013,
"count": 1,
"is_parallel": true,
"self": 0.0006129929997769068,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.0012392710000312945,
"count": 8,
"is_parallel": true,
"self": 0.0012392710000312945
}
}
},
"UnityEnvironment.step": {
"total": 0.03517509899984361,
"count": 1,
"is_parallel": true,
"self": 0.00033470899961685063,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.0003099690002272837,
"count": 1,
"is_parallel": true,
"self": 0.0003099690002272837
},
"communicator.exchange": {
"total": 0.03348615799995969,
"count": 1,
"is_parallel": true,
"self": 0.03348615799995969
},
"steps_from_proto": {
"total": 0.0010442630000397912,
"count": 1,
"is_parallel": true,
"self": 0.00025534899987178505,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.0007889140001680062,
"count": 8,
"is_parallel": true,
"self": 0.0007889140001680062
}
}
}
}
}
}
},
"UnityEnvironment.step": {
"total": 15.73382117801384,
"count": 1490,
"is_parallel": true,
"self": 0.4480130580141122,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.3202367619951474,
"count": 1490,
"is_parallel": true,
"self": 0.3202367619951474
},
"communicator.exchange": {
"total": 13.62650583200275,
"count": 1490,
"is_parallel": true,
"self": 13.62650583200275
},
"steps_from_proto": {
"total": 1.339065526001832,
"count": 1490,
"is_parallel": true,
"self": 0.28065136999066453,
"children": {
"_process_rank_one_or_two_observation": {
"total": 1.0584141560111675,
"count": 11920,
"is_parallel": true,
"self": 1.0584141560111675
}
}
}
}
}
}
}
}
}
}
},
"trainer_advance": {
"total": 12.86194856900147,
"count": 1491,
"self": 0.03044504200306619,
"children": {
"process_trajectory": {
"total": 2.4885163209983148,
"count": 1491,
"self": 2.4885163209983148
},
"_update_policy": {
"total": 10.342987206000089,
"count": 5,
"self": 5.611499314000866,
"children": {
"TorchPPOOptimizer.update": {
"total": 4.7314878919992225,
"count": 471,
"self": 4.7314878919992225
}
}
}
}
}
}
},
"trainer_threads": {
"total": 1.1469996934465598e-06,
"count": 1,
"self": 1.1469996934465598e-06
},
"TrainerController._save_models": {
"total": 0.11070096299999932,
"count": 1,
"self": 0.0011625470001490612,
"children": {
"RLTrainer._checkpoint": {
"total": 0.10953841599985026,
"count": 1,
"self": 0.10953841599985026
}
}
}
}
}
}
}