krisha-n's picture
First Push
27b9efe
{
"name": "root",
"metadata": {
"timer_format_version": "0.1.0",
"start_time_seconds": "1714471886",
"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": "1714471897"
},
"total": 10.520304202000034,
"count": 1,
"self": 0.42746481700004324,
"children": {
"run_training.setup": {
"total": 0.05404926599999271,
"count": 1,
"self": 0.05404926599999271
},
"TrainerController.start_learning": {
"total": 10.038790118999998,
"count": 1,
"self": 0.023395055999799297,
"children": {
"TrainerController._reset_env": {
"total": 3.4098076289999995,
"count": 1,
"self": 3.4098076289999995
},
"TrainerController.advance": {
"total": 6.523799687000235,
"count": 209,
"self": 0.003084671000635808,
"children": {
"env_step": {
"total": 6.520715015999599,
"count": 209,
"self": 3.0125868259996196,
"children": {
"SubprocessEnvManager._take_step": {
"total": 3.5049756210000282,
"count": 209,
"self": 0.016911653999386544,
"children": {
"TorchPolicy.evaluate": {
"total": 3.4880639670006417,
"count": 209,
"self": 3.4880639670006417
}
}
},
"workers": {
"total": 0.0031525689999511997,
"count": 209,
"self": 0.0,
"children": {
"worker_root": {
"total": 9.983392315999993,
"count": 209,
"is_parallel": true,
"self": 7.475502062999794,
"children": {
"run_training.setup": {
"total": 0.0,
"count": 0,
"is_parallel": true,
"self": 0.0,
"children": {
"steps_from_proto": {
"total": 0.006787582000015391,
"count": 1,
"is_parallel": true,
"self": 0.0034861029999433413,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.0033014790000720495,
"count": 10,
"is_parallel": true,
"self": 0.0033014790000720495
}
}
},
"UnityEnvironment.step": {
"total": 0.03592902200000481,
"count": 1,
"is_parallel": true,
"self": 0.0007374909999953161,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.0004473200000347788,
"count": 1,
"is_parallel": true,
"self": 0.0004473200000347788
},
"communicator.exchange": {
"total": 0.032669510000005175,
"count": 1,
"is_parallel": true,
"self": 0.032669510000005175
},
"steps_from_proto": {
"total": 0.0020747009999695365,
"count": 1,
"is_parallel": true,
"self": 0.0003904829998759851,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.0016842180000935514,
"count": 10,
"is_parallel": true,
"self": 0.0016842180000935514
}
}
}
}
}
}
},
"UnityEnvironment.step": {
"total": 2.5078902530001983,
"count": 208,
"is_parallel": true,
"self": 0.11758743600000798,
"children": {
"UnityEnvironment._generate_step_input": {
"total": 0.06396938199975466,
"count": 208,
"is_parallel": true,
"self": 0.06396938199975466
},
"communicator.exchange": {
"total": 1.9354172840000956,
"count": 208,
"is_parallel": true,
"self": 1.9354172840000956
},
"steps_from_proto": {
"total": 0.39091615100034005,
"count": 208,
"is_parallel": true,
"self": 0.07716464599963047,
"children": {
"_process_rank_one_or_two_observation": {
"total": 0.3137515050007096,
"count": 2080,
"is_parallel": true,
"self": 0.3137515050007096
}
}
}
}
}
}
}
}
}
}
}
}
},
"trainer_threads": {
"total": 2.6390999948944227e-05,
"count": 1,
"self": 2.6390999948944227e-05,
"children": {
"thread_root": {
"total": 0.0,
"count": 0,
"is_parallel": true,
"self": 0.0,
"children": {
"trainer_advance": {
"total": 6.461062212003469,
"count": 11190,
"is_parallel": true,
"self": 0.16997438300160184,
"children": {
"process_trajectory": {
"total": 3.7529145130018264,
"count": 11190,
"is_parallel": true,
"self": 3.223184624001874,
"children": {
"RLTrainer._checkpoint": {
"total": 0.5297298889999524,
"count": 4,
"is_parallel": true,
"self": 0.5297298889999524
}
}
},
"_update_policy": {
"total": 2.538173316000041,
"count": 1,
"is_parallel": true,
"self": 0.26403752000015857,
"children": {
"TorchPPOOptimizer.update": {
"total": 2.2741357959998822,
"count": 33,
"is_parallel": true,
"self": 2.2741357959998822
}
}
}
}
}
}
}
}
},
"TrainerController._save_models": {
"total": 0.08176135600001544,
"count": 1,
"self": 0.0007255700000428078,
"children": {
"RLTrainer._checkpoint": {
"total": 0.08103578599997263,
"count": 1,
"self": 0.08103578599997263
}
}
}
}
}
}
}