KrishBakshi commited on
Commit
22d75bd
·
verified ·
1 Parent(s): 4a6fe3b
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: ml-agents
3
+ tags:
4
+ - SnowballTarget
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - ML-Agents-SnowballTarget
8
+ ---
9
+
10
+ # **ppo** Agent playing **SnowballTarget**
11
+ This is a trained model of a **ppo** agent playing **SnowballTarget**
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-SnowballTarget
33
+ 3. Step 2: Select your *.nn /*.onnx file
34
+ 4. Click on Watch the agent play 👀
35
+
SnowballTarget.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:993bb6a3e5e3fc4c05af2795d5228d6e6182ee2e8871666e763bf3312b617a3a
3
+ size 650646
SnowballTarget/SnowballTarget-149984.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b06d6387e4770d6d71d53b57903dab51d9212ecc62e29225e9bf419aacd1232
3
+ size 650646
SnowballTarget/SnowballTarget-149984.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eec1ec9d717881669b62e1bd887c3ffcf6f0587d61dbf6a8b44e063c2444eb93
3
+ size 3849710
SnowballTarget/SnowballTarget-199984.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:993bb6a3e5e3fc4c05af2795d5228d6e6182ee2e8871666e763bf3312b617a3a
3
+ size 650646
SnowballTarget/SnowballTarget-199984.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a7cec822d1867b01e36b106c122cfcef69585ab214a7ea92b37efa83769d508
3
+ size 3849710
SnowballTarget/SnowballTarget-200112.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:993bb6a3e5e3fc4c05af2795d5228d6e6182ee2e8871666e763bf3312b617a3a
3
+ size 650646
SnowballTarget/SnowballTarget-200112.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0841f76ae8c5a9d56a309d78a8a90f50a9a9f5432492b71a9c8a324aca7ce68
3
+ size 3849710
SnowballTarget/SnowballTarget-49936.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9eae9c7f037e62822bb41d87889944cf14dad10191f2b863204de51fa7146684
3
+ size 650646
SnowballTarget/SnowballTarget-49936.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe76aaebf4b280fdda46184497fd6b4a3960bb4588b90263864ff071dd94da06
3
+ size 3849635
SnowballTarget/SnowballTarget-99960.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd96e8d9f9658dfb984dcc34d8b83def05aaaa08b959d67d8b5e81b31dd6c2b7
3
+ size 650646
SnowballTarget/SnowballTarget-99960.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62734a8da9b292ea5f351834e1781c58836e400748b8726dac39da0b5017c2db
3
+ size 3849635
SnowballTarget/checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62b31b221bb543114a27fac3acbde48dbf86e8e5e702a5115dd682ada972c86d
3
+ size 3848821
SnowballTarget/events.out.tfevents.1781362262.archlinux.82894.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b926bcc7689f2ffd1221273719b38e5bab630e4c2ef219daf594ed910ef5517b
3
+ size 28169
SnowballTarget/events.out.tfevents.1781363167.archlinux.86520.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6b623ae90e3d6dddc3da44704c8b3153b4a20c8da8519b239481526459281a2
3
+ size 1114
SnowballTarget/events.out.tfevents.1781363233.archlinux.87022.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dca5e4140c79b29f724903fc081a604bc017e83080bbee96c22f8c7a709e03f5
3
+ size 1114
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"default_settings": null, "behaviors": {"SnowballTarget": {"trainer_type": "ppo", "hyperparameters": {"batch_size": 128, "buffer_size": 2048, "learning_rate": 0.0003, "beta": 0.005, "epsilon": 0.2, "lambd": 0.95, "num_epoch": 3, "shared_critic": false, "learning_rate_schedule": "linear", "beta_schedule": "linear", "epsilon_schedule": "linear"}, "checkpoint_interval": 50000, "network_settings": {"normalize": false, "hidden_units": 256, "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}}}, "init_path": null, "keep_checkpoints": 10, "even_checkpoints": false, "max_steps": 200000, "time_horizon": 64, "summary_freq": 10000, "threaded": true, "self_play": null, "behavioral_cloning": null}}, "env_settings": {"env_path": "./training-envs-executables/linux/SnowballTarget/SnowballTarget.x86_64", "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": "SnowballTarget_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,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_settings: null
2
+ behaviors:
3
+ SnowballTarget:
4
+ trainer_type: ppo
5
+ hyperparameters:
6
+ batch_size: 128
7
+ buffer_size: 2048
8
+ learning_rate: 0.0003
9
+ beta: 0.005
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: 50000
18
+ network_settings:
19
+ normalize: false
20
+ hidden_units: 256
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
+ init_path: null
39
+ keep_checkpoints: 10
40
+ even_checkpoints: false
41
+ max_steps: 200000
42
+ time_horizon: 64
43
+ summary_freq: 10000
44
+ threaded: true
45
+ self_play: null
46
+ behavioral_cloning: null
47
+ env_settings:
48
+ env_path: ./training-envs-executables/linux/SnowballTarget/SnowballTarget.x86_64
49
+ env_args:
50
+ - -screen-fullscreen
51
+ - '1'
52
+ base_port: 5005
53
+ num_envs: 1
54
+ num_areas: 1
55
+ timeout_wait: 60
56
+ seed: -1
57
+ max_lifetime_restarts: 10
58
+ restarts_rate_limit_n: 1
59
+ restarts_rate_limit_period_s: 60
60
+ engine_settings:
61
+ width: 1280
62
+ height: 720
63
+ quality_level: 5
64
+ time_scale: 1.0
65
+ target_frame_rate: 30
66
+ capture_frame_rate: 30
67
+ no_graphics: false
68
+ no_graphics_monitor: false
69
+ environment_parameters: null
70
+ checkpoint_settings:
71
+ run_id: SnowballTarget_v1
72
+ initialize_from: null
73
+ load_model: false
74
+ resume: true
75
+ force: false
76
+ train_model: false
77
+ inference: true
78
+ results_dir: results
79
+ torch_settings:
80
+ device: null
81
+ debug: false
run_logs/Player-0.log ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Mono path[0] = '/home/krish/wrk/ml-agents/training-envs-executables/linux/SnowballTarget/SnowballTarget_Data/Managed'
2
+ Mono config path = '/home/krish/wrk/ml-agents/training-envs-executables/linux/SnowballTarget/SnowballTarget_Data/MonoBleedingEdge/etc'
3
+ Preloaded 'lib_burst_generated.so'
4
+ Preloaded 'libgrpc_csharp_ext.x64.so'
5
+ Display 0 'screen': 1280x720 (primary device).
6
+ Desktop is 1280 x 720 @ 0 Hz
7
+ Initialize engine version: 2021.3.14f1 (eee1884e7226)
8
+ [Subsystems] Discovering subsystems at path /home/krish/wrk/ml-agents/training-envs-executables/linux/SnowballTarget/SnowballTarget_Data/UnitySubsystems
9
+ GfxDevice: creating device client; threaded=1; jobified=1
10
+ Renderer: llvmpipe (LLVM 22.1.1, 256 bits)
11
+ Vendor: Mesa
12
+ Version: 4.5 (Core Profile) Mesa 26.0.3-arch1.1
13
+ GLES: 0
14
+ 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
15
+ 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
16
+ _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
17
+ 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
18
+ 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
19
+ 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
20
+ R_multiview GL_OVR_multiview2 GL_S3_s3tc
21
+ OPENGL LOG: Creating OpenGL 4.5 graphics device ; Context level <OpenGL 4.5> ; Context handle 989579488
22
+ Begin MonoManager ReloadAssembly
23
+ - Completed reload, in 0.050 seconds
24
+ Default vsync count 1
25
+ requesting resize 1280 x 720
26
+ Using native desktop resolution 1280 x 720
27
+ requesting fullscreen 1280 x 720 at 0 Hz
28
+ Desktop is 1280 x 720 @ 0 Hz
29
+ UnloadTime: 0.771927 ms
30
+ requesting resize 1280 x 720
31
+ resizing window to 1280 x 720
32
+ Desktop is 1280 x 720 @ 0 Hz
33
+ X connection to :99 broken (explicit kill or server shutdown).
run_logs/timers.json ADDED
@@ -0,0 +1,197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "root",
3
+ "metadata": {
4
+ "timer_format_version": "0.1.0",
5
+ "start_time_seconds": "1781363229",
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/SnowballTarget.yaml --env=./training-envs-executables/linux/SnowballTarget/SnowballTarget.x86_64 --run-id=SnowballTarget_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": "1781363310"
14
+ },
15
+ "total": 80.77528322899889,
16
+ "count": 1,
17
+ "self": 0.12479214800987393,
18
+ "children": {
19
+ "run_training.setup": {
20
+ "total": 0.013791589997708797,
21
+ "count": 1,
22
+ "self": 0.013791589997708797
23
+ },
24
+ "TrainerController.start_learning": {
25
+ "total": 80.6366994909913,
26
+ "count": 1,
27
+ "self": 0.007458304215106182,
28
+ "children": {
29
+ "TrainerController._reset_env": {
30
+ "total": 4.358114005997777,
31
+ "count": 1,
32
+ "self": 4.358114005997777
33
+ },
34
+ "TrainerController.advance": {
35
+ "total": 76.27109974977793,
36
+ "count": 301,
37
+ "self": 0.003090769052505493,
38
+ "children": {
39
+ "env_step": {
40
+ "total": 76.26800898072543,
41
+ "count": 301,
42
+ "self": 75.20546286167519,
43
+ "children": {
44
+ "SubprocessEnvManager._take_step": {
45
+ "total": 1.0593237821740331,
46
+ "count": 301,
47
+ "self": 0.018899253249401227,
48
+ "children": {
49
+ "TorchPolicy.evaluate": {
50
+ "total": 1.0404245289246319,
51
+ "count": 301,
52
+ "self": 1.0404245289246319
53
+ }
54
+ }
55
+ },
56
+ "workers": {
57
+ "total": 0.0032223368762061,
58
+ "count": 300,
59
+ "self": 0.0,
60
+ "children": {
61
+ "worker_root": {
62
+ "total": 80.56040642608423,
63
+ "count": 300,
64
+ "is_parallel": true,
65
+ "self": 6.449326069079689,
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.0011906779982382432,
75
+ "count": 1,
76
+ "is_parallel": true,
77
+ "self": 0.00032490900775883347,
78
+ "children": {
79
+ "_process_rank_one_or_two_observation": {
80
+ "total": 0.0008657689904794097,
81
+ "count": 10,
82
+ "is_parallel": true,
83
+ "self": 0.0008657689904794097
84
+ }
85
+ }
86
+ },
87
+ "UnityEnvironment.step": {
88
+ "total": 0.5368912950070808,
89
+ "count": 1,
90
+ "is_parallel": true,
91
+ "self": 0.00023171199427451938,
92
+ "children": {
93
+ "UnityEnvironment._generate_step_input": {
94
+ "total": 0.00020793899602722377,
95
+ "count": 1,
96
+ "is_parallel": true,
97
+ "self": 0.00020793899602722377
98
+ },
99
+ "communicator.exchange": {
100
+ "total": 0.535729662005906,
101
+ "count": 1,
102
+ "is_parallel": true,
103
+ "self": 0.535729662005906
104
+ },
105
+ "steps_from_proto": {
106
+ "total": 0.0007219820108730346,
107
+ "count": 1,
108
+ "is_parallel": true,
109
+ "self": 0.0001563119876664132,
110
+ "children": {
111
+ "_process_rank_one_or_two_observation": {
112
+ "total": 0.0005656700232066214,
113
+ "count": 10,
114
+ "is_parallel": true,
115
+ "self": 0.0005656700232066214
116
+ }
117
+ }
118
+ }
119
+ }
120
+ }
121
+ }
122
+ },
123
+ "UnityEnvironment.step": {
124
+ "total": 74.11108035700454,
125
+ "count": 299,
126
+ "is_parallel": true,
127
+ "self": 0.08205600488872733,
128
+ "children": {
129
+ "UnityEnvironment._generate_step_input": {
130
+ "total": 0.06866319198161364,
131
+ "count": 299,
132
+ "is_parallel": true,
133
+ "self": 0.06866319198161364
134
+ },
135
+ "communicator.exchange": {
136
+ "total": 73.72095986509521,
137
+ "count": 299,
138
+ "is_parallel": true,
139
+ "self": 73.72095986509521
140
+ },
141
+ "steps_from_proto": {
142
+ "total": 0.23940129503898788,
143
+ "count": 299,
144
+ "is_parallel": true,
145
+ "self": 0.05124291285756044,
146
+ "children": {
147
+ "_process_rank_one_or_two_observation": {
148
+ "total": 0.18815838218142744,
149
+ "count": 2990,
150
+ "is_parallel": true,
151
+ "self": 0.18815838218142744
152
+ }
153
+ }
154
+ }
155
+ }
156
+ }
157
+ }
158
+ }
159
+ }
160
+ }
161
+ }
162
+ }
163
+ }
164
+ },
165
+ "trainer_threads": {
166
+ "total": 2.7431000489741564e-05,
167
+ "count": 1,
168
+ "self": 2.7431000489741564e-05,
169
+ "children": {
170
+ "thread_root": {
171
+ "total": 0.0,
172
+ "count": 0,
173
+ "is_parallel": true,
174
+ "self": 0.0,
175
+ "children": {
176
+ "trainer_advance": {
177
+ "total": 75.65314266071073,
178
+ "count": 159972,
179
+ "is_parallel": true,
180
+ "self": 0.7389369632437592,
181
+ "children": {
182
+ "process_trajectory": {
183
+ "total": 74.91420569746697,
184
+ "count": 159972,
185
+ "is_parallel": true,
186
+ "self": 74.91420569746697
187
+ }
188
+ }
189
+ }
190
+ }
191
+ }
192
+ }
193
+ }
194
+ }
195
+ }
196
+ }
197
+ }
run_logs/training_status.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "SnowballTarget": {
3
+ "checkpoints": [
4
+ {
5
+ "steps": 49936,
6
+ "file_path": "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-49936.onnx",
7
+ "reward": 12.545454545454545,
8
+ "creation_time": 1781362351.1090949,
9
+ "auxillary_file_paths": [
10
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-49936.pt"
11
+ ]
12
+ },
13
+ {
14
+ "steps": 99960,
15
+ "file_path": "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-99960.onnx",
16
+ "reward": 18.454545454545453,
17
+ "creation_time": 1781362440.1677294,
18
+ "auxillary_file_paths": [
19
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-99960.pt"
20
+ ]
21
+ },
22
+ {
23
+ "steps": 149984,
24
+ "file_path": "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-149984.onnx",
25
+ "reward": 25.181818181818183,
26
+ "creation_time": 1781362529.951121,
27
+ "auxillary_file_paths": [
28
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-149984.pt"
29
+ ]
30
+ },
31
+ {
32
+ "steps": 199984,
33
+ "file_path": "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-199984.onnx",
34
+ "reward": 25.363636363636363,
35
+ "creation_time": 1781362619.8554373,
36
+ "auxillary_file_paths": [
37
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-199984.pt"
38
+ ]
39
+ },
40
+ {
41
+ "steps": 200112,
42
+ "file_path": "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-200112.onnx",
43
+ "reward": 25.363636363636363,
44
+ "creation_time": 1781362620.0212786,
45
+ "auxillary_file_paths": [
46
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-200112.pt"
47
+ ]
48
+ }
49
+ ],
50
+ "final_checkpoint": {
51
+ "steps": 200112,
52
+ "file_path": "results/SnowballTarget_v1/SnowballTarget.onnx",
53
+ "reward": 25.363636363636363,
54
+ "creation_time": 1781362620.0212786,
55
+ "auxillary_file_paths": [
56
+ "results/SnowballTarget_v1/SnowballTarget/SnowballTarget-200112.pt"
57
+ ]
58
+ }
59
+ },
60
+ "metadata": {
61
+ "stats_format_version": "0.3.0",
62
+ "mlagents_version": "1.2.0.dev0",
63
+ "torch_version": "2.8.0+cu128"
64
+ }
65
+ }