KrishBakshi commited on
Commit
bee144a
·
verified ·
1 Parent(s): f912f41

Add PyramidsRND PPO trained model

Browse files
Pyramids.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ab2087143bc425411b4688abc09ededfb5f432b85fd7b9ee6a5f8f7523b5e77
3
+ size 1420238
Pyramids/Pyramids-1000007.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ab2087143bc425411b4688abc09ededfb5f432b85fd7b9ee6a5f8f7523b5e77
3
+ size 1420238
Pyramids/Pyramids-1000007.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:911e88dee101f5254c091d4d0f93619eb7d79e62c442acdc5a3021a766b4f987
3
+ size 8656079
Pyramids/Pyramids-499919.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e28067ad910bc7e967a4b2a72707c2863f3ba145eeb16c23ab8deb0bfcf8da5
3
+ size 1420238
Pyramids/Pyramids-499919.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e02e9b8613fa17de4eb024125cdd5134eae2547204e442bc053f22a7db1f8125
3
+ size 8656000
Pyramids/Pyramids-999879.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ab2087143bc425411b4688abc09ededfb5f432b85fd7b9ee6a5f8f7523b5e77
3
+ size 1420238
Pyramids/Pyramids-999879.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee232abd2b77bcbc4f78fa8f9d68c6c49622b10cda017302b580a573686658c4
3
+ size 8656000
Pyramids/checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:373428581a418a48850f1187c54a84812357f478c8840a11ebc2fc7a1d8427fa
3
+ size 8655605
Pyramids/events.out.tfevents.1781433660.archlinux.124999.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0b01f8fcf0633169c5bb2c5b78c048ae66da0175e967a15db2011144a2a464f
3
+ size 314252
Pyramids/events.out.tfevents.1781434906.archlinux.129560.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3449a9a9f8bf7cef0ec2dedc0aec60bae6e8a328826f46fdd01242c9d37ecf6a
3
+ size 1423
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: ml-agents
3
+ tags:
4
+ - Pyramids
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - ML-Agents-Pyramids
8
+ ---
9
+
10
+ # **ppo** Agent playing **Pyramids**
11
+ This is a trained model of a **ppo** agent playing **Pyramids**
12
+ using the [Unity ML-Agents Library](https://github.com/Unity-Technologies/ml-agents).
13
+
14
+ ## Usage (with ML-Agents)
15
+ The Documentation: https://unity-technologies.github.io/ml-agents/ML-Agents-Toolkit-Documentation/
16
+
17
+ We wrote a complete tutorial to learn to train your first agent using ML-Agents and publish it to the Hub:
18
+ - A *short tutorial* where you teach Huggy the Dog 🐶 to fetch the stick and then play with him directly in your
19
+ browser: https://huggingface.co/learn/deep-rl-course/unitbonus1/introduction
20
+ - A *longer tutorial* to understand how works ML-Agents:
21
+ https://huggingface.co/learn/deep-rl-course/unit5/introduction
22
+
23
+ ### Resume the training
24
+ ```bash
25
+ mlagents-learn <your_configuration_file_path.yaml> --run-id=<run_id> --resume
26
+ ```
27
+
28
+ ### Watch your Agent play
29
+ You can watch your agent **playing directly in your browser**
30
+
31
+ 1. If the environment is part of ML-Agents official environments, go to https://huggingface.co/unity
32
+ 2. Step 1: Find your model_id: KrishBakshi/ppo-Pyramids
33
+ 3. Step 2: Select your *.nn /*.onnx file
34
+ 4. Click on Watch the agent play 👀
35
+
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"default_settings": null, "behaviors": {"Pyramids": {"trainer_type": "ppo", "hyperparameters": {"batch_size": 128, "buffer_size": 2048, "learning_rate": 0.0003, "beta": 0.01, "epsilon": 0.2, "lambd": 0.95, "num_epoch": 3, "shared_critic": false, "learning_rate_schedule": "linear", "beta_schedule": "linear", "epsilon_schedule": "linear"}, "checkpoint_interval": 500000, "network_settings": {"normalize": false, "hidden_units": 512, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "reward_signals": {"extrinsic": {"gamma": 0.99, "strength": 1.0, "network_settings": {"normalize": false, "hidden_units": 128, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}}, "rnd": {"gamma": 0.99, "strength": 0.01, "network_settings": {"normalize": false, "hidden_units": 64, "num_layers": 3, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "learning_rate": 0.0001, "encoding_size": null}}, "init_path": null, "keep_checkpoints": 5, "even_checkpoints": false, "max_steps": 1000000, "time_horizon": 128, "summary_freq": 30000, "threaded": false, "self_play": null, "behavioral_cloning": null}}, "env_settings": {"env_path": "./training-envs-executables/Pyramids/Pyramids/Pyramids", "env_args": ["-screen-fullscreen", "1"], "base_port": 5005, "num_envs": 1, "num_areas": 1, "timeout_wait": 60, "seed": -1, "max_lifetime_restarts": 10, "restarts_rate_limit_n": 1, "restarts_rate_limit_period_s": 60}, "engine_settings": {"width": 1280, "height": 720, "quality_level": 5, "time_scale": 1.0, "target_frame_rate": 30, "capture_frame_rate": 30, "no_graphics": false, "no_graphics_monitor": false}, "environment_parameters": null, "checkpoint_settings": {"run_id": "PyramidsRND_v1", "initialize_from": null, "load_model": false, "resume": true, "force": false, "train_model": false, "inference": true, "results_dir": "results"}, "torch_settings": {"device": null}, "debug": false}
configuration.yaml ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_settings: null
2
+ behaviors:
3
+ Pyramids:
4
+ trainer_type: ppo
5
+ hyperparameters:
6
+ batch_size: 128
7
+ buffer_size: 2048
8
+ learning_rate: 0.0003
9
+ beta: 0.01
10
+ epsilon: 0.2
11
+ lambd: 0.95
12
+ num_epoch: 3
13
+ shared_critic: false
14
+ learning_rate_schedule: linear
15
+ beta_schedule: linear
16
+ epsilon_schedule: linear
17
+ checkpoint_interval: 500000
18
+ network_settings:
19
+ normalize: false
20
+ hidden_units: 512
21
+ num_layers: 2
22
+ vis_encode_type: simple
23
+ memory: null
24
+ goal_conditioning_type: hyper
25
+ deterministic: false
26
+ reward_signals:
27
+ extrinsic:
28
+ gamma: 0.99
29
+ strength: 1.0
30
+ network_settings:
31
+ normalize: false
32
+ hidden_units: 128
33
+ num_layers: 2
34
+ vis_encode_type: simple
35
+ memory: null
36
+ goal_conditioning_type: hyper
37
+ deterministic: false
38
+ rnd:
39
+ gamma: 0.99
40
+ strength: 0.01
41
+ network_settings:
42
+ normalize: false
43
+ hidden_units: 64
44
+ num_layers: 3
45
+ vis_encode_type: simple
46
+ memory: null
47
+ goal_conditioning_type: hyper
48
+ deterministic: false
49
+ learning_rate: 0.0001
50
+ encoding_size: null
51
+ init_path: null
52
+ keep_checkpoints: 5
53
+ even_checkpoints: false
54
+ max_steps: 1000000
55
+ time_horizon: 128
56
+ summary_freq: 30000
57
+ threaded: false
58
+ self_play: null
59
+ behavioral_cloning: null
60
+ env_settings:
61
+ env_path: ./training-envs-executables/Pyramids/Pyramids/Pyramids
62
+ env_args:
63
+ - -screen-fullscreen
64
+ - '1'
65
+ base_port: 5005
66
+ num_envs: 1
67
+ num_areas: 1
68
+ timeout_wait: 60
69
+ seed: -1
70
+ max_lifetime_restarts: 10
71
+ restarts_rate_limit_n: 1
72
+ restarts_rate_limit_period_s: 60
73
+ engine_settings:
74
+ width: 1280
75
+ height: 720
76
+ quality_level: 5
77
+ time_scale: 1.0
78
+ target_frame_rate: 30
79
+ capture_frame_rate: 30
80
+ no_graphics: false
81
+ no_graphics_monitor: false
82
+ environment_parameters: null
83
+ checkpoint_settings:
84
+ run_id: PyramidsRND_v1
85
+ initialize_from: null
86
+ load_model: false
87
+ resume: true
88
+ force: false
89
+ train_model: false
90
+ inference: true
91
+ results_dir: results
92
+ torch_settings:
93
+ device: null
94
+ debug: false
run_logs/Player-0.log ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Mono path[0] = '/home/krish/wrk/ml-agents/training-envs-executables/Pyramids/Pyramids/Pyramids_Data/Managed'
2
+ Mono config path = '/home/krish/wrk/ml-agents/training-envs-executables/Pyramids/Pyramids/Pyramids_Data/MonoBleedingEdge/etc'
3
+ Found 2 interfaces on host : 0) 192.168.1.15 1) 192.168.1.12
4
+ Multi-casting "[IP] 192.168.1.15 [Port] 55310 [Flags] 2 [Guid] 308980634 [EditorId] 764847374 [Version] 1048832 [Id] LinuxPlayer(13,192.168.1.15) [Debug] 0 [PackageName] LinuxPlayer [ProjectName] UnityEnvironment" to [225.0.0.222:54997]...
5
+ Preloaded 'lib_burst_generated.so'
6
+ Preloaded 'libgrpc_csharp_ext.x64.so'
7
+ Display 0 'screen': 1280x720 (primary device).
8
+ Desktop is 1280 x 720 @ 0 Hz
9
+ Initialize engine version: 2021.3.5f1 (40eb3a945986)
10
+ [Subsystems] Discovering subsystems at path /home/krish/wrk/ml-agents/training-envs-executables/Pyramids/Pyramids/Pyramids_Data/UnitySubsystems
11
+ GfxDevice: creating device client; threaded=1; jobified=0
12
+ Renderer: llvmpipe (LLVM 22.1.1, 256 bits)
13
+ Vendor: Mesa
14
+ Version: 4.5 (Core Profile) Mesa 26.0.3-arch1.1
15
+ GLES: 0
16
+ GL_3DFX_texture_compression_FXT1 GL_AMD_conservative_depth GL_AMD_draw_buffers_blend GL_AMD_gpu_shader_int64 GL_AMD_multi_draw_indirect GL_AMD_pinned_memory GL_AMD_query_buffer_object GL_AMD_seamless_cubemap_per_texture GL_AMD_shader_stencil_export GL_AMD_shader_trinary_minmax GL_AMD_texture_texture4 GL_AMD_vertex_shader_layer GL_AMD_vertex_shader_viewport_index GL_ANGLE_texture_compression_dxt3 GL_ANGLE_texture_compression_dxt5 GL_ARB_ES2_compatibility GL_ARB_ES3_1_compatibility GL_ARB_ES3_2_compatibility GL_ARB_ES3_compatibility GL_ARB_arrays_of_arrays GL_ARB_base_instance GL_ARB_blend_func_extended GL_ARB_buffer_storage GL_ARB_clear_buffer_object GL_ARB_clear_texture GL_ARB_clip_control GL_ARB_compressed_texture_pixel_storage GL_ARB_compute_shader GL_ARB_conditional_render_inverted GL_ARB_conservative_depth GL_ARB_copy_buffer GL_ARB_copy_image GL_ARB_cull_distance GL_ARB_debug_output GL_ARB_depth_buffer_float GL_ARB_depth_clamp GL_ARB_derivative_control GL_ARB_direct_state_access GL_ARB_draw_buffers GL_A
17
+ RB_draw_buffers_blend GL_ARB_draw_elements_base_vertex GL_ARB_draw_indirect GL_ARB_draw_instanced GL_ARB_enhanced_layouts GL_ARB_explicit_attrib_location GL_ARB_explicit_uniform_location GL_ARB_fragment_coord_conventions GL_ARB_fragment_layer_viewport GL_ARB_fragment_shader GL_ARB_framebuffer_no_attachments GL_ARB_framebuffer_object GL_ARB_framebuffer_sRGB GL_ARB_get_program_binary GL_ARB_get_texture_sub_image GL_ARB_gl_spirv GL_ARB_gpu_shader5 GL_ARB_gpu_shader_fp64 GL_ARB_gpu_shader_int64 GL_ARB_half_float_pixel GL_ARB_half_float_vertex GL_ARB_indirect_parameters GL_ARB_instanced_arrays GL_ARB_internalformat_query GL_ARB_internalformat_query2 GL_ARB_invalidate_subdata GL_ARB_map_buffer_alignment GL_ARB_map_buffer_range GL_ARB_multi_bind GL_ARB_multi_draw_indirect GL_ARB_occlusion_query2 GL_ARB_parallel_shader_compile GL_ARB_pipeline_statistics_query GL_ARB_pixel_buffer_object GL_ARB_point_sprite GL_ARB_polygon_offset_clamp GL_ARB_post_depth_coverage GL_ARB_program_interface_query GL_ARB_provoking_vertex GL
18
+ _ARB_query_buffer_object GL_ARB_robust_buffer_access_behavior GL_ARB_robustness GL_ARB_sample_locations GL_ARB_sample_shading GL_ARB_sampler_objects GL_ARB_seamless_cube_map GL_ARB_seamless_cubemap_per_texture GL_ARB_separate_shader_objects GL_ARB_shader_atomic_counter_ops GL_ARB_shader_atomic_counters GL_ARB_shader_ballot GL_ARB_shader_bit_encoding GL_ARB_shader_clock GL_ARB_shader_draw_parameters GL_ARB_shader_group_vote GL_ARB_shader_image_load_store GL_ARB_shader_image_size GL_ARB_shader_objects GL_ARB_shader_precision GL_ARB_shader_stencil_export GL_ARB_shader_storage_buffer_object GL_ARB_shader_subroutine GL_ARB_shader_texture_image_samples GL_ARB_shader_texture_lod GL_ARB_shader_viewport_layer_array GL_ARB_shading_language_420pack GL_ARB_shading_language_include GL_ARB_shading_language_packing GL_ARB_spirv_extensions GL_ARB_stencil_texturing GL_ARB_sync GL_ARB_tessellation_shader GL_ARB_texture_barrier GL_ARB_texture_border_clamp GL_ARB_texture_buffer_object GL_ARB_texture_buffer_object_rgb32 GL_ARB_t
19
+ exture_buffer_range GL_ARB_texture_compression_bptc GL_ARB_texture_compression_rgtc GL_ARB_texture_cube_map_array GL_ARB_texture_filter_anisotropic GL_ARB_texture_filter_minmax GL_ARB_texture_float GL_ARB_texture_gather GL_ARB_texture_mirror_clamp_to_edge GL_ARB_texture_multisample GL_ARB_texture_non_power_of_two GL_ARB_texture_query_levels GL_ARB_texture_query_lod GL_ARB_texture_rectangle GL_ARB_texture_rg GL_ARB_texture_rgb10_a2ui GL_ARB_texture_stencil8 GL_ARB_texture_storage GL_ARB_texture_storage_multisample GL_ARB_texture_swizzle GL_ARB_texture_view GL_ARB_timer_query GL_ARB_transform_feedback2 GL_ARB_transform_feedback3 GL_ARB_transform_feedback_instanced GL_ARB_transform_feedback_overflow_query GL_ARB_uniform_buffer_object GL_ARB_vertex_array_bgra GL_ARB_vertex_array_object GL_ARB_vertex_attrib_64bit GL_ARB_vertex_attrib_binding GL_ARB_vertex_buffer_object GL_ARB_vertex_shader GL_ARB_vertex_type_10f_11f_11f_rev GL_ARB_vertex_type_2_10_10_10_rev GL_ARB_viewport_array GL_ARM_shader_framebuffer_fetch_de
20
+ pth_stencil GL_ATI_blend_equation_separate GL_ATI_meminfo GL_ATI_texture_float GL_ATI_texture_mirror_once GL_EXT_EGL_image_storage GL_EXT_EGL_image_storage_compression GL_EXT_EGL_sync GL_EXT_abgr GL_EXT_blend_equation_separate GL_EXT_debug_label GL_EXT_depth_bounds_test GL_EXT_draw_buffers2 GL_EXT_draw_instanced GL_EXT_framebuffer_blit GL_EXT_framebuffer_multisample GL_EXT_framebuffer_multisample_blit_scaled GL_EXT_framebuffer_object GL_EXT_framebuffer_sRGB GL_EXT_memory_object GL_EXT_memory_object_fd GL_EXT_packed_depth_stencil GL_EXT_packed_float GL_EXT_pixel_buffer_object GL_EXT_polygon_offset_clamp GL_EXT_provoking_vertex GL_EXT_shader_clock GL_EXT_shader_framebuffer_fetch GL_EXT_shader_framebuffer_fetch_non_coherent GL_EXT_shader_image_load_formatted GL_EXT_shader_integer_mix GL_EXT_texture_array GL_EXT_texture_compression_dxt1 GL_EXT_texture_compression_rgtc GL_EXT_texture_compression_s3tc GL_EXT_texture_filter_anisotropic GL_EXT_texture_filter_minmax GL_EXT_texture_integer GL_EXT_texture_mirror_clamp
21
+ GL_EXT_texture_sRGB GL_EXT_texture_sRGB_R8 GL_EXT_texture_sRGB_RG8 GL_EXT_texture_sRGB_decode GL_EXT_texture_shadow_lod GL_EXT_texture_shared_exponent GL_EXT_texture_snorm GL_EXT_texture_storage GL_EXT_texture_swizzle GL_EXT_timer_query GL_EXT_transform_feedback GL_EXT_vertex_array_bgra GL_EXT_vertex_attrib_64bit GL_IBM_multimode_draw_arrays GL_INTEL_shader_atomic_float_minmax GL_KHR_blend_equation_advanced GL_KHR_blend_equation_advanced_coherent GL_KHR_context_flush_control GL_KHR_debug GL_KHR_no_error GL_KHR_parallel_shader_compile GL_KHR_robust_buffer_access_behavior GL_KHR_robustness GL_KHR_texture_compression_astc_ldr GL_KHR_texture_compression_astc_sliced_3d GL_MESA_framebuffer_flip_y GL_MESA_pack_invert GL_MESA_shader_integer_functions GL_MESA_texture_signed_rgba GL_NVX_gpu_memory_info GL_NV_alpha_to_coverage_dither_control GL_NV_conditional_render GL_NV_copy_image GL_NV_depth_clamp GL_NV_packed_depth_stencil GL_NV_sample_locations GL_NV_shader_atomic_float GL_NV_texture_barrier GL_OES_EGL_image GL_OV
22
+ R_multiview GL_OVR_multiview2 GL_S3_s3tc
23
+ OPENGL LOG: Creating OpenGL 4.5 graphics device ; Context level <OpenGL 4.5> ; Context handle 991563424
24
+ Begin MonoManager ReloadAssembly
25
+ - Completed reload, in 0.054 seconds
26
+ Default vsync count 1
27
+ requesting resize 1280 x 720
28
+ Using native desktop resolution 1280 x 720
29
+ requesting fullscreen 1280 x 720 at 0 Hz
30
+ Desktop is 1280 x 720 @ 0 Hz
31
+ UnloadTime: 0.855434 ms
32
+ Setting up 8 worker threads for Enlighten.
33
+ requesting resize 1280 x 720
34
+ resizing window to 1280 x 720
35
+ Desktop is 1280 x 720 @ 0 Hz
36
+ X connection to :99 broken (explicit kill or server shutdown).
run_logs/timers.json ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "root",
3
+ "metadata": {
4
+ "timer_format_version": "0.1.0",
5
+ "start_time_seconds": "1781434903",
6
+ "python_version": "3.10.12 (main, Jul 5 2023, 18:54:27) [GCC 11.2.0]",
7
+ "command_line_arguments": "/home/krish/miniconda3/envs/rl/bin/mlagents-learn ./config/ppo/PyramidsRND.yaml --env=./training-envs-executables/Pyramids/Pyramids/Pyramids --run-id=PyramidsRND_v1 --inference --resume --width=1280 --height=720 --time-scale=1 --target-frame-rate=30 --capture-frame-rate=30 --env-args -screen-fullscreen 1",
8
+ "mlagents_version": "1.2.0.dev0",
9
+ "mlagents_envs_version": "1.2.0.dev0",
10
+ "communication_protocol_version": "1.5.0",
11
+ "pytorch_version": "2.8.0+cu128",
12
+ "numpy_version": "1.23.5",
13
+ "end_time_seconds": "1781434987"
14
+ },
15
+ "total": 83.93240248897928,
16
+ "count": 1,
17
+ "self": 0.04536691302200779,
18
+ "children": {
19
+ "run_training.setup": {
20
+ "total": 0.014802956982748583,
21
+ "count": 1,
22
+ "self": 0.014802956982748583
23
+ },
24
+ "TrainerController.start_learning": {
25
+ "total": 83.87223261897452,
26
+ "count": 1,
27
+ "self": 0.03179568875930272,
28
+ "children": {
29
+ "TrainerController._reset_env": {
30
+ "total": 4.284268685005372,
31
+ "count": 1,
32
+ "self": 4.284268685005372
33
+ },
34
+ "TrainerController.advance": {
35
+ "total": 79.55616718321107,
36
+ "count": 1183,
37
+ "self": 0.02702596431481652,
38
+ "children": {
39
+ "env_step": {
40
+ "total": 77.83172014908632,
41
+ "count": 1183,
42
+ "self": 74.95250500791008,
43
+ "children": {
44
+ "SubprocessEnvManager._take_step": {
45
+ "total": 2.8602947444014717,
46
+ "count": 1183,
47
+ "self": 0.08303508427343331,
48
+ "children": {
49
+ "TorchPolicy.evaluate": {
50
+ "total": 2.7772596601280384,
51
+ "count": 1149,
52
+ "self": 2.7772596601280384
53
+ }
54
+ }
55
+ },
56
+ "workers": {
57
+ "total": 0.018920396774774417,
58
+ "count": 1182,
59
+ "self": 0.0,
60
+ "children": {
61
+ "worker_root": {
62
+ "total": 83.80282710303436,
63
+ "count": 1182,
64
+ "is_parallel": true,
65
+ "self": 11.380317655974068,
66
+ "children": {
67
+ "run_training.setup": {
68
+ "total": 0.0,
69
+ "count": 0,
70
+ "is_parallel": true,
71
+ "self": 0.0,
72
+ "children": {
73
+ "steps_from_proto": {
74
+ "total": 0.0011298650060780346,
75
+ "count": 1,
76
+ "is_parallel": true,
77
+ "self": 0.00031689307070337236,
78
+ "children": {
79
+ "_process_rank_one_or_two_observation": {
80
+ "total": 0.0008129719353746623,
81
+ "count": 8,
82
+ "is_parallel": true,
83
+ "self": 0.0008129719353746623
84
+ }
85
+ }
86
+ },
87
+ "UnityEnvironment.step": {
88
+ "total": 0.11441023799125105,
89
+ "count": 1,
90
+ "is_parallel": true,
91
+ "self": 0.0003504969645291567,
92
+ "children": {
93
+ "UnityEnvironment._generate_step_input": {
94
+ "total": 0.00021565300994552672,
95
+ "count": 1,
96
+ "is_parallel": true,
97
+ "self": 0.00021565300994552672
98
+ },
99
+ "communicator.exchange": {
100
+ "total": 0.11249973101075739,
101
+ "count": 1,
102
+ "is_parallel": true,
103
+ "self": 0.11249973101075739
104
+ },
105
+ "steps_from_proto": {
106
+ "total": 0.0013443570060189813,
107
+ "count": 1,
108
+ "is_parallel": true,
109
+ "self": 0.00021412002388387918,
110
+ "children": {
111
+ "_process_rank_one_or_two_observation": {
112
+ "total": 0.0011302369821351022,
113
+ "count": 8,
114
+ "is_parallel": true,
115
+ "self": 0.0011302369821351022
116
+ }
117
+ }
118
+ }
119
+ }
120
+ }
121
+ }
122
+ },
123
+ "UnityEnvironment.step": {
124
+ "total": 72.42250944706029,
125
+ "count": 1181,
126
+ "is_parallel": true,
127
+ "self": 0.4916796865290962,
128
+ "children": {
129
+ "UnityEnvironment._generate_step_input": {
130
+ "total": 0.3221646789170336,
131
+ "count": 1181,
132
+ "is_parallel": true,
133
+ "self": 0.3221646789170336
134
+ },
135
+ "communicator.exchange": {
136
+ "total": 70.39682772691594,
137
+ "count": 1181,
138
+ "is_parallel": true,
139
+ "self": 70.39682772691594
140
+ },
141
+ "steps_from_proto": {
142
+ "total": 1.2118373546982184,
143
+ "count": 1181,
144
+ "is_parallel": true,
145
+ "self": 0.2656794083013665,
146
+ "children": {
147
+ "_process_rank_one_or_two_observation": {
148
+ "total": 0.9461579463968519,
149
+ "count": 9448,
150
+ "is_parallel": true,
151
+ "self": 0.9461579463968519
152
+ }
153
+ }
154
+ }
155
+ }
156
+ }
157
+ }
158
+ }
159
+ }
160
+ }
161
+ }
162
+ },
163
+ "trainer_advance": {
164
+ "total": 1.6974210698099341,
165
+ "count": 1182,
166
+ "self": 0.0619773285870906,
167
+ "children": {
168
+ "process_trajectory": {
169
+ "total": 1.6354437412228435,
170
+ "count": 1182,
171
+ "self": 1.6354437412228435
172
+ }
173
+ }
174
+ }
175
+ }
176
+ },
177
+ "trainer_threads": {
178
+ "total": 1.0619987733662128e-06,
179
+ "count": 1,
180
+ "self": 1.0619987733662128e-06
181
+ }
182
+ }
183
+ }
184
+ }
185
+ }
run_logs/training_status.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "Pyramids": {
3
+ "checkpoints": [
4
+ {
5
+ "steps": 499919,
6
+ "file_path": "results/PyramidsRND_v1/Pyramids/Pyramids-499919.onnx",
7
+ "reward": null,
8
+ "creation_time": 1781434213.5734684,
9
+ "auxillary_file_paths": [
10
+ "results/PyramidsRND_v1/Pyramids/Pyramids-499919.pt"
11
+ ]
12
+ },
13
+ {
14
+ "steps": 999879,
15
+ "file_path": "results/PyramidsRND_v1/Pyramids/Pyramids-999879.onnx",
16
+ "reward": 1.6865000029404957,
17
+ "creation_time": 1781434829.881059,
18
+ "auxillary_file_paths": [
19
+ "results/PyramidsRND_v1/Pyramids/Pyramids-999879.pt"
20
+ ]
21
+ },
22
+ {
23
+ "steps": 1000007,
24
+ "file_path": "results/PyramidsRND_v1/Pyramids/Pyramids-1000007.onnx",
25
+ "reward": 1.6865000029404957,
26
+ "creation_time": 1781434829.9428637,
27
+ "auxillary_file_paths": [
28
+ "results/PyramidsRND_v1/Pyramids/Pyramids-1000007.pt"
29
+ ]
30
+ }
31
+ ],
32
+ "final_checkpoint": {
33
+ "steps": 1000007,
34
+ "file_path": "results/PyramidsRND_v1/Pyramids.onnx",
35
+ "reward": 1.6865000029404957,
36
+ "creation_time": 1781434829.9428637,
37
+ "auxillary_file_paths": [
38
+ "results/PyramidsRND_v1/Pyramids/Pyramids-1000007.pt"
39
+ ]
40
+ }
41
+ },
42
+ "metadata": {
43
+ "stats_format_version": "0.3.0",
44
+ "mlagents_version": "1.2.0.dev0",
45
+ "torch_version": "2.8.0+cu128"
46
+ }
47
+ }