KenanKhan's picture
First commit
0764751 verified
{
"name": "root",
"metadata": {
"timer_format_version": "0.1.0",
"start_time_seconds": "1715938292",
"python_version": "3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0]",
"command_line_arguments": "/usr/local/bin/mlagents-learn ./config/ppo/SnowballTarget.yaml --env=./training-envs-executables/linux/SnowballTarget/SnowballTarget --run-id=SnowballTarget1 --no-graphics",
"mlagents_version": "1.1.0.dev0",
"mlagents_envs_version": "1.1.0.dev0",
"communication_protocol_version": "1.5.0",
"pytorch_version": "2.2.1+cu121",
"numpy_version": "1.23.5",
"end_time_seconds": "1715938309"
},
"total": 16.618466336999973,
"count": 1,
"self": 0.4925311999999167,
"children": {
"run_training.setup": {
"total": 0.06570098000008784,
"count": 1,
"self": 0.06570098000008784
},
"TrainerController.start_learning": {
"total": 16.060234156999968,
"count": 1,
"self": 0.026919756999973288,
"children": {
"TrainerController._reset_env": {
"total": 2.6446715119999453,
"count": 1,
"self": 2.6446715119999453
},
"TrainerController.advance": {
"total": 13.301549367000007,
"count": 476,
"self": 0.009383547998709219,
"children": {
"env_step": {
"total": 13.292165819001298,
"count": 476,
"self": 10.346704423000347,
"children": {
"SubprocessEnvManager._take_step": {
"total": 2.9364452710002524,
"count": 476,
"self": 0.05062360700071622,
"children": {
"TorchPolicy.evaluate": {
"total": 2.885821663999536,
"count": 476,
"self": 2.885821663999536
}
}
},
"workers": {
"total": 0.009016125000698594,
"count": 476,
"self": 0.0,
"children": {
"worker_root": {
"total": 15.981206916998872,
"count": 476,
"is_parallel": true,
"self": 8.706304038000212,
"children": {
"run_training.setup": {
"total": 0.0,
"count": 0,
"is_parallel": true,
"self": 0.0,
"children": {
"steps_from_proto": {
"total": 0.005568052000057833,
"count": 1,
"is_parallel": true,
"self": 0.0038171580001744587,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.001750893999883374,
"count": 10,
"is_parallel": true,
"self": 0.001750893999883374
}
}
},
"UnityEnvironment.step": {
"total": 0.03848917999994228,
"count": 1,
"is_parallel": true,
"self": 0.0008206499999232619,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.00040637200004312035,
"count": 1,
"is_parallel": true,
"self": 0.00040637200004312035
},
"communicator.exchange": {
"total": 0.03471340800001599,
"count": 1,
"is_parallel": true,
"self": 0.03471340800001599
},
"steps_from_proto": {
"total": 0.0025487499999599095,
"count": 1,
"is_parallel": true,
"self": 0.0006087699998715834,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.001939980000088326,
"count": 10,
"is_parallel": true,
"self": 0.001939980000088326
}
}
}
}
}
}
},
"UnityEnvironment.step": {
"total": 7.27490287899866,
"count": 475,
"is_parallel": true,
"self": 0.3126263209945819,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.17117439300159276,
"count": 475,
"is_parallel": true,
"self": 0.17117439300159276
},
"communicator.exchange": {
"total": 5.785309041000687,
"count": 475,
"is_parallel": true,
"self": 5.785309041000687
},
"steps_from_proto": {
"total": 1.0057931240017979,
"count": 475,
"is_parallel": true,
"self": 0.19957870200335037,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.8062144219984475,
"count": 4750,
"is_parallel": true,
"self": 0.8062144219984475
}
}
}
}
}
}
}
}
}
}
}
}
},
"trainer_threads": {
"total": 3.151400005663163e-05,
"count": 1,
"self": 3.151400005663163e-05,
"children": {
"thread_root": {
"total": 0.0,
"count": 0,
"is_parallel": true,
"self": 0.0,
"children": {
"trainer_advance": {
"total": 13.174855514007277,
"count": 17061,
"is_parallel": true,
"self": 0.3275742720004473,
"children": {
"process_trajectory": {
"total": 8.172808581006734,
"count": 17061,
"is_parallel": true,
"self": 6.946078679006973,
"children": {
"RLTrainer._checkpoint": {
"total": 1.2267299019997608,
"count": 5,
"is_parallel": true,
"self": 1.2267299019997608
}
}
},
"_update_policy": {
"total": 4.674472661000095,
"count": 2,
"is_parallel": true,
"self": 1.4532365219998837,
"children": {
"TorchPPOOptimizer.update": {
"total": 3.2212361390002116,
"count": 99,
"is_parallel": true,
"self": 3.2212361390002116
}
}
}
}
}
}
}
}
},
"TrainerController._save_models": {
"total": 0.0870620069999859,
"count": 1,
"self": 0.001118453000003683,
"children": {
"RLTrainer._checkpoint": {
"total": 0.08594355399998221,
"count": 1,
"self": 0.08594355399998221
}
}
}
}
}
}
}